neuralworm commited on
Commit
866c3f0
·
1 Parent(s): d15bd24

update tests

Browse files
cognitive_mapping_probe/signal_analysis.py CHANGED
@@ -73,4 +73,4 @@ def get_power_spectrum_for_plotting(state_deltas: np.ndarray) -> Tuple[np.ndarra
73
  xf = rfftfreq(n, 1.0)
74
 
75
  power_spectrum = np.abs(yf)**2
76
- return xf, power_spectrum
 
73
  xf = rfftfreq(n, 1.0)
74
 
75
  power_spectrum = np.abs(yf)**2
76
+ return xf, power_spectrum
tests/test_components.py CHANGED
@@ -82,4 +82,4 @@ def test_analyze_cognitive_signal_with_multiple_peaks():
82
 
83
  assert results["dominant_periods_steps"] is not None
84
  expected_periods = [10.0, 4.0, 30.0]
85
- assert results["dominant_periods_steps"] == expected_periods
 
82
 
83
  assert results["dominant_periods_steps"] is not None
84
  expected_periods = [10.0, 4.0, 30.0]
85
+ assert results["dominant_periods_steps"] == expected_periods