From 4334b47b461fa9be15d2a11be7fba7fc81172afe Mon Sep 17 00:00:00 2001
From: su530201 <olivier.kaufmann@umons.ac.be>
Date: Sat, 22 Apr 2023 16:47:19 +0200
Subject: [PATCH] Fixes a bug in vab_square_wave

---
 measure.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/measure.py b/measure.py
index e1888c7e..3ec42d48 100644
--- 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):
-- 
GitLab