diff --git a/src/spikegadgets_to_nwb/tests/test_convert_dios.py b/src/spikegadgets_to_nwb/tests/test_convert_dios.py index 2cc0455..773a670 100644 --- 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