Test dio timestamps

LorenFrankLab · Aug 20, 2023 · e8450e1 · e8450e1
1 parent 388b82b
commit e8450e1
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/spikegadgets_to_nwb/tests/test_convert_dios.py b/src/spikegadgets_to_nwb/tests/test_convert_dios.py
@@ -55,9 +55,12 @@ def test_add_dios_single_rec():
                 ]
                 # check that timeseries match
                 np.testing.assert_array_equal(current_dio.data[:], old_dio.data[:])
-                # np.testing.assert_array_equal(
-                #     current_dio.timestamps, old_dio.timestamps
-                # )
+                assert np.allclose(
+                    current_dio.timestamps[:],
+                    old_dio.timestamps[:],
+                    rtol=0,
+                    atol=1.0 / 30000,
+                )
                 assert current_dio.unit == old_dio.unit
                 assert current_dio.description == old_dio.description
 
@@ -117,7 +120,12 @@ def test_add_dios_two_epoch():
                 ]
                 # check that timeseries match
                 np.testing.assert_array_equal(current_dio.data, old_dio.data)
-                # np.testing.assert_array_equal(current_dio.timestamps, old_dio.timestamps)  # TODO uncomment
+                assert np.allclose(
+                    current_dio.timestamps[:],
+                    old_dio.timestamps[:],
+                    rtol=0,
+                    atol=1.0 / 30000,
+                )
                 assert current_dio.unit == old_dio.unit
                 assert current_dio.description == old_dio.description