Adapt tests

cta-observatory · Nov 22, 2023 · d247152 · d247152
1 parent a51316f
commit d247152
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lstchain/tools/tests/test_create_drs4_pedestal_file.py b/lstchain/tools/tests/test_create_drs4_pedestal_file.py
@@ -35,12 +35,12 @@ def test_create_drs4_pedestal_file(tmp_path):
     baseline_mean = drs4_data['baseline_mean']
     baseline_counts = drs4_data['baseline_std']
 
-    assert baseline_mean.dtype == np.uint16
+    assert baseline_mean.dtype == np.int16
     assert baseline_mean.shape == (N_GAINS, N_PIXELS, N_CAPACITORS_PIXEL)
     assert np.isclose(np.average(baseline_mean[baseline_counts > 0], weights=baseline_counts[baseline_counts > 0]), 400, rtol=0.05)
 
     spike_height = drs4_data['spike_height']
-    assert spike_height.dtype == np.uint16
+    assert spike_height.dtype == np.int16
     mean_spike_height = np.nanmean(spike_height, axis=(0, 1))
 
     # these are the expected spike heights, but due to the low statistics,