set raw_timestamps=True when reading files

adamreeve · Jan 24, 2024 · d2bbddd · d2bbddd
1 parent 5cd13a9
commit d2bbddd
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/nptdms/test/test_example_files.py b/nptdms/test/test_example_files.py
@@ -68,12 +68,14 @@ def test_defragment(tdms_file):
     # verify we can defragment a file with raw timestamps
     TdmsWriter.defragment(test_file_path, output_file)
 
-    # rewind output file BytesIO instance and read it back in as a TdmsFile
+    # rewind output file BytesIO instance, so it can read it back in as a TdmsFile
     output_file.seek(0)
-    output_tdms = tdms.TdmsFile(output_file)
 
     # verify that both TdmsFile objects are the same
-    tdms_files_assert_equal(tdms.TdmsFile(test_file_path), output_tdms)
+    tdms_files_assert_equal(
+        tdms.TdmsFile(test_file_path, raw_timestamps=True),
+        tdms.TdmsFile(output_file, raw_timestamps=True),
+    )
 
 
 def test_big_endian_format():