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
|