Skip to content

Commit bc471dc

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 1fbd4ce commit bc471dc

5 files changed

Lines changed: 95 additions & 95 deletions

File tree

tests/unit/test_analytics.py

Lines changed: 24 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,9 @@ def test_fill_false_blocks(
3939

4040
result = sleep_detection._fill_false_blocks(sleep_idx_array, gap_block)
4141

42-
assert np.array_equal(
43-
result, expected_result
44-
), f"Expected {expected_result}, but got {result}"
42+
assert np.array_equal(result, expected_result), (
43+
f"Expected {expected_result}, but got {result}"
44+
)
4545

4646

4747
def test_compute_abs_diff_mean_anglez(
@@ -58,9 +58,9 @@ def test_compute_abs_diff_mean_anglez(
5858

5959
result = sleep_detection._compute_abs_diff_mean_anglez(sleep_detection.anglez)
6060

61-
assert np.array_equal(
62-
result.measurements, expected_result
63-
), f"Expected {expected_result}, but got {result.measurements}"
61+
assert np.array_equal(result.measurements, expected_result), (
62+
f"Expected {expected_result}, but got {result.measurements}"
63+
)
6464
assert np.array_equal(len(result.time), expected_length)
6565

6666

@@ -161,12 +161,12 @@ def test_spt_window(sleep_detection: analytics.GgirSleepDetection) -> None:
161161

162162
result = sleep_detection._spt_window(sleep_detection.anglez)
163163

164-
assert np.array_equal(
165-
result.measurements, expected_result
166-
), f"Expected {expected_result}, but got {result.measurements}"
167-
assert np.array_equal(
168-
len(result.time), expected_length
169-
), f"Expected {expected_length}, but got {len(result.time)}"
164+
assert np.array_equal(result.measurements, expected_result), (
165+
f"Expected {expected_result}, but got {result.measurements}"
166+
)
167+
assert np.array_equal(len(result.time), expected_length), (
168+
f"Expected {expected_length}, but got {len(result.time)}"
169+
)
170170

171171

172172
def test_spt_window_null(sleep_detection: analytics.GgirSleepDetection) -> None:
@@ -176,12 +176,12 @@ def test_spt_window_null(sleep_detection: analytics.GgirSleepDetection) -> None:
176176

177177
result = sleep_detection._spt_window(sleep_detection.anglez)
178178

179-
assert np.array_equal(
180-
result.measurements, expected_result
181-
), f"Expected {expected_result}, but got {result.measurements}"
182-
assert np.array_equal(
183-
len(result.time), expected_length
184-
), f"Expected {expected_length}, but got {len(result.time)}"
179+
assert np.array_equal(result.measurements, expected_result), (
180+
f"Expected {expected_result}, but got {result.measurements}"
181+
)
182+
assert np.array_equal(len(result.time), expected_length), (
183+
f"Expected {expected_length}, but got {len(result.time)}"
184+
)
185185

186186

187187
def test_calculate_sib_periods(sleep_detection: analytics.GgirSleepDetection) -> None:
@@ -191,12 +191,12 @@ def test_calculate_sib_periods(sleep_detection: analytics.GgirSleepDetection) ->
191191

192192
result = sleep_detection._calculate_sib_periods(sleep_detection.anglez, 10)
193193

194-
assert np.array_equal(
195-
result.measurements, expected_result
196-
), f"Expected {expected_result}, but got {result.measurements}"
197-
assert (
198-
len(result.measurements) == expected_length
199-
), f"Expected {expected_length}, but got {len(result.measurements)}"
194+
assert np.array_equal(result.measurements, expected_result), (
195+
f"Expected {expected_result}, but got {result.measurements}"
196+
)
197+
assert len(result.measurements) == expected_length, (
198+
f"Expected {expected_length}, but got {len(result.measurements)}"
199+
)
200200

201201

202202
def test_find_onset_wakeup_times(sleep_detection: analytics.GgirSleepDetection) -> None:

tests/unit/test_calibration.py

Lines changed: 36 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -115,12 +115,12 @@ def test_closest_point_fit() -> None:
115115

116116
linear_transform = calibrator._closest_point_fit((unit_sphere * scale) + offset)
117117

118-
assert np.allclose(
119-
linear_transform.scale, expected_scale, atol=1e-3
120-
), f"Scale is {linear_transform.scale} expected {expected_scale}"
121-
assert np.allclose(
122-
linear_transform.offset, expected_offset, atol=1e-3
123-
), f"Offset is {linear_transform.offset} expected {expected_offset})"
118+
assert np.allclose(linear_transform.scale, expected_scale, atol=1e-3), (
119+
f"Scale is {linear_transform.scale} expected {expected_scale}"
120+
)
121+
assert np.allclose(linear_transform.offset, expected_offset, atol=1e-3), (
122+
f"Offset is {linear_transform.offset} expected {expected_offset})"
123+
)
124124

125125

126126
def test_closest_point_fit_constrainedmin() -> None:
@@ -136,12 +136,12 @@ def test_closest_point_fit_constrainedmin() -> None:
136136

137137
linear_transform = calibrator._closest_point_fit((unit_sphere * scale) + offset)
138138

139-
assert np.allclose(
140-
linear_transform.scale, expected_scale, atol=1e-3
141-
), f"Scale is {linear_transform.scale} expected {expected_scale}"
142-
assert np.allclose(
143-
linear_transform.offset, expected_offset, atol=1e-3
144-
), f"Offset is {linear_transform.offset} expected {expected_offset})"
139+
assert np.allclose(linear_transform.scale, expected_scale, atol=1e-3), (
140+
f"Scale is {linear_transform.scale} expected {expected_scale}"
141+
)
142+
assert np.allclose(linear_transform.offset, expected_offset, atol=1e-3), (
143+
f"Offset is {linear_transform.offset} expected {expected_offset})"
144+
)
145145

146146

147147
def test_closest_point_fit_constrainmin_calibration_error() -> None:
@@ -201,12 +201,12 @@ def test_ggir_calibration_successful() -> None:
201201

202202
linear_transform = calibrator._calibrate(dummy_measure)
203203

204-
assert np.allclose(
205-
linear_transform.scale, expected_scale, atol=1e-3
206-
), f"Scale is {linear_transform.scale} expected {expected_scale}"
207-
assert np.allclose(
208-
linear_transform.offset, expected_offset, atol=1e-3
209-
), f"Offset is {linear_transform.offset} expected {expected_offset}"
204+
assert np.allclose(linear_transform.scale, expected_scale, atol=1e-3), (
205+
f"Scale is {linear_transform.scale} expected {expected_scale}"
206+
)
207+
assert np.allclose(linear_transform.offset, expected_offset, atol=1e-3), (
208+
f"Offset is {linear_transform.offset} expected {expected_offset}"
209+
)
210210

211211

212212
@pytest.mark.parametrize(
@@ -275,12 +275,12 @@ def test_run_ggircalibration() -> None:
275275

276276
result = calibrator.run_calibration(dummy_measure)
277277

278-
assert isinstance(
279-
result, models.Measurement
280-
), f"was expecting type models.Measurement, object is of type {type(result)}"
281-
assert np.allclose(
282-
result.measurements, expected_data.measurements, atol=1e-3
283-
), "Measurement data did not match"
278+
assert isinstance(result, models.Measurement), (
279+
f"was expecting type models.Measurement, object is of type {type(result)}"
280+
)
281+
assert np.allclose(result.measurements, expected_data.measurements, atol=1e-3), (
282+
"Measurement data did not match"
283+
)
284284
assert result.time.equals(expected_data.time), "Time series are not equal"
285285

286286

@@ -308,12 +308,12 @@ def test_run_constrainedmincalibration() -> None:
308308

309309
result = calibrator.run_calibration(dummy_measure)
310310

311-
assert isinstance(
312-
result, models.Measurement
313-
), f"was expecting type models.Measurement, object is of type {type(result)}"
314-
assert np.allclose(
315-
result.measurements, expected_data.measurements, atol=1e-3
316-
), "Measurement data did not match"
311+
assert isinstance(result, models.Measurement), (
312+
f"was expecting type models.Measurement, object is of type {type(result)}"
313+
)
314+
assert np.allclose(result.measurements, expected_data.measurements, atol=1e-3), (
315+
"Measurement data did not match"
316+
)
317317
assert result.time.equals(expected_data.time), "Time series are not equal"
318318

319319

@@ -341,10 +341,10 @@ def test_run_chunked_calibration() -> None:
341341

342342
result = calibrator.run_calibration(dummy_measure)
343343

344-
assert isinstance(
345-
result, models.Measurement
346-
), f"was expecting type models.Measurement, object is of type {type(result)}"
347-
assert np.allclose(
348-
result.measurements, expected_data.measurements, atol=1e-3
349-
), "Measurement data did not match"
344+
assert isinstance(result, models.Measurement), (
345+
f"was expecting type models.Measurement, object is of type {type(result)}"
346+
)
347+
assert np.allclose(result.measurements, expected_data.measurements, atol=1e-3), (
348+
"Measurement data did not match"
349+
)
350350
assert result.time.equals(expected_data.time), "Time series are not equal"

tests/unit/test_cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,6 @@ def test_main_with_bad_epoch(
133133
"""Test cli with invalid epoch length."""
134134
with pytest.raises(
135135
ValueError,
136-
match="Value for epoch_length is:-5." "Please enter an integer >= 0.",
136+
match="Value for epoch_length is:-5.Please enter an integer >= 0.",
137137
):
138138
cli.main([str(sample_data_gt3x), "-e", "-5"])

tests/unit/test_computations.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -181,6 +181,6 @@ def test_moving_median(window_size: int, expected_output: np.ndarray) -> None:
181181
f"measurements array are not the same shape. Expected {expected_output.shape}, "
182182
f"instead got: {test_result.measurements.shape}"
183183
)
184-
assert np.all(
185-
np.isclose(test_result.measurements, expected_output)
186-
), "Test results do not match the expected output"
184+
assert np.all(np.isclose(test_result.measurements, expected_output)), (
185+
"Test results do not match the expected output"
186+
)

tests/unit/test_metrics.py

Lines changed: 31 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -54,17 +54,17 @@ def test_euclidean_norm_minus_one(
5454

5555
enmo_results = metrics.euclidean_norm_minus_one(test_acceleration)
5656

57-
assert np.all(
58-
np.isclose(enmo_results.measurements, expected_enmo)
59-
), f"Expected {expected_enmo}"
57+
assert np.all(np.isclose(enmo_results.measurements, expected_enmo)), (
58+
f"Expected {expected_enmo}"
59+
)
6060

61-
assert enmo_results.time.equals(
62-
test_acceleration.time
63-
), "Input time attribute does not match output time attribute."
61+
assert enmo_results.time.equals(test_acceleration.time), (
62+
"Input time attribute does not match output time attribute."
63+
)
6464

65-
assert enmo_results.measurements.shape == (
66-
TEST_LENGTH,
67-
), f"Expected enmo shape: ({TEST_LENGTH},), got ({enmo_results.measurements.shape})"
65+
assert enmo_results.measurements.shape == (TEST_LENGTH,), (
66+
f"Expected enmo shape: ({TEST_LENGTH},), got ({enmo_results.measurements.shape})"
67+
)
6868

6969

7070
@pytest.mark.parametrize(
@@ -93,12 +93,12 @@ def test_angle_relative_to_horizontal(
9393
np.isclose(angle_z_results.measurements, expected_anglez, equal_nan=True)
9494
), f"Expected {expected_anglez}, got: {angle_z_results.measurements}"
9595

96-
assert angle_z_results.time.equals(
97-
test_acceleration.time
98-
), "Input time attribute does not match output time attribute."
96+
assert angle_z_results.time.equals(test_acceleration.time), (
97+
"Input time attribute does not match output time attribute."
98+
)
9999

100100
assert angle_z_results.measurements.shape == (1,), (
101-
f"Expected anglez shape: {(1,)}," f"got({angle_z_results.measurements.shape})"
101+
f"Expected anglez shape: {(1,)},got({angle_z_results.measurements.shape})"
102102
)
103103

104104

@@ -121,9 +121,9 @@ def test_cleanup_isolated_ones_nonwear_value(
121121
"""Test the cleanup isolated ones nonwear value function."""
122122
test_result = metrics._cleanup_isolated_ones_nonwear_value(nonwear_value_array)
123123

124-
assert np.all(
125-
test_result == expected_result
126-
), f"Expected {expected_result}, got: {test_result}"
124+
assert np.all(test_result == expected_result), (
125+
f"Expected {expected_result}, got: {test_result}"
126+
)
127127

128128

129129
def test_group_acceleration_data_by_time() -> None:
@@ -164,9 +164,9 @@ def test_compute_nonwear_value_per_axis(
164164
acceleration_grouped["X"], std_criteria=modifier
165165
)
166166

167-
assert (
168-
test_resultx == expected_result
169-
), f"Expected {expected_result}, got: {test_resultx}"
167+
assert test_resultx == expected_result, (
168+
f"Expected {expected_result}, got: {test_resultx}"
169+
)
170170

171171

172172
def test_compute_nonwear_value_array(create_acceleration: pl.DataFrame) -> None:
@@ -185,12 +185,12 @@ def test_compute_nonwear_value_array(create_acceleration: pl.DataFrame) -> None:
185185
std_criteria=1,
186186
)
187187

188-
assert np.all(
189-
test_result == expected_result
190-
), f"Expected {expected_result}, got: {test_result}"
191-
assert (
192-
len(test_result) == expected_time_length
193-
), f"Expected time to be {expected_time_length}, got: {len(test_result)}"
188+
assert np.all(test_result == expected_result), (
189+
f"Expected {expected_result}, got: {test_result}"
190+
)
191+
assert len(test_result) == expected_time_length, (
192+
f"Expected time to be {expected_time_length}, got: {len(test_result)}"
193+
)
194194

195195

196196
@pytest.mark.parametrize(
@@ -220,9 +220,9 @@ def test_detect_nonwear(
220220
std_criteria=modifier,
221221
)
222222

223-
assert np.all(
224-
test_result.measurements == modifier
225-
), f"Expected non-wear flag value to be {expected_result}, got: {test_result}"
226-
assert (
227-
len(test_result.time) == expected_time_length
228-
), f"Expected time to be {expected_time_length}, got: {len(test_result.time)}"
223+
assert np.all(test_result.measurements == modifier), (
224+
f"Expected non-wear flag value to be {expected_result}, got: {test_result}"
225+
)
226+
assert len(test_result.time) == expected_time_length, (
227+
f"Expected time to be {expected_time_length}, got: {len(test_result.time)}"
228+
)

0 commit comments

Comments
 (0)