Update test_analytics.py

Asanto32 · Asanto32 · commit 80eb0dded5c2 · 2025-04-16T13:51:44.000-04:00
Fix organization of test
diff --git a/tests/unit/test_analytics.py b/tests/unit/test_analytics.py
@@ -214,11 +214,12 @@ def test_sleep_cleanup() -> None:
         )
     ]
 
+    expected_result = np.zeros(len(nonwear_measurement.time))
+    expected_result[600:2000] = 1
+
     result = analytics.sleep_cleanup(
         sleep_windows=sleep_windows, nonwear_measurement=nonwear_measurement
     )
-    expected_result = np.zeros(len(nonwear_measurement.time))
-    expected_result[600:2000] = 1
 
     assert len(result.time) == 3600
     assert np.array_equal(result.measurements, expected_result)

Original file line number	Diff line number	Diff line change
`@@ -214,11 +214,12 @@ def test_sleep_cleanup() -> None:`
`214`	`214`	`)`
`215`	`215`	`]`
`216`	`216`
	`217`	`+ expected_result = np.zeros(len(nonwear_measurement.time))`
	`218`	`+ expected_result[600:2000] = 1`
	`219`	`+`
`217`	`220`	`result = analytics.sleep_cleanup(`
`218`	`221`	`sleep_windows=sleep_windows, nonwear_measurement=nonwear_measurement`
`219`	`222`	`)`
`220`		`- expected_result = np.zeros(len(nonwear_measurement.time))`
`221`		`- expected_result[600:2000] = 1`
`222`	`223`
`223`	`224`	`assert len(result.time) == 3600`
`224`	`225`	`assert np.array_equal(result.measurements, expected_result)`