Fixes a bug in vab_square_wave

4334b47b · Olivier Kaufmann · c2f91f04 · 4334b47b
Commit 4334b47b authored 1 year ago by Olivier Kaufmann
Hide whitespace changes
Inline Side-by-side

Showing

with 2 additions and 1 deletion
+2 -1
--- a/measure.py
+++ b/measure.py
@@ -151,7 +151,8 @@ class OhmPiHardware:
            polarity = 1
        return vab, polarity, rab

-    def vab_square_wave(self, vab, length, sampling_rate, cycles=3):
+    def vab_square_wave(self, vab, length, sampling_rate, cycles=3, polarity=1):
+        self.tx.polarity = polarity
        self._vab_pulses(vab, [length]*cycles, sampling_rate)

    def _vab_pulse(self, vab, length, sampling_rate=None, polarity=None, append=False):