test fix

mariarv · Oct 4, 2023 · bcb60c3 · bcb60c3
1 parent 5253ed5
commit bcb60c3
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/efel/pyfeatures/pyfeatures.py b/efel/pyfeatures/pyfeatures.py
@@ -82,7 +82,9 @@ def impedance():
             else:
                 Z = fft_volt / fft_cur
                 norm_Z = abs(Z) / max(abs(Z))
-                select_idxs = numpy.swapaxes(numpy.argwhere((freq > 0) & (freq <= Z_max_freq)), 0, 1)[0]
+                select_idxs = numpy.swapaxes(
+                    numpy.argwhere((freq > 0) & (freq <= Z_max_freq)), 0, 1
+                )[0]
                 smooth_Z = gaussian_filter1d(norm_Z[select_idxs], 10)
                 ind_max = numpy.argmax(smooth_Z)
                 return freq[ind_max]

diff --git a/efel/units/units.json b/efel/units/units.json
@@ -143,5 +143,6 @@
     "initburst_sahp": "mV",
     "initburst_sahp_ssse": "mV",
     "initburst_sahp_vb": "mV",
-    "depol_block_bool": "constant"
+    "depol_block_bool": "constant",
+    "impedance": "Hz"
 }
diff --git a/tests/test_pyfeatures.py b/tests/test_pyfeatures.py
@@ -383,6 +383,5 @@ def test_impedance():
 
     feature_name = "impedance"
 
-    expected_values = {feature_name: 4.23076923076923}
+    expected_values = {feature_name: 4.615384615384615}
     _test_expected_value(feature_name, expected_values)
-    assert False