test(core)(measurement): Simplified tests in test_experiment.py for saving too fast test cases to: use repeat scan for fastest subsequent experiments, only run expts 3 times, ensure the time stamp is the same for all three, ensure that the names are incremented as expected, and ensure that the file names are the same.

Author: rsbrost

test/measurement/test_experiment.py

@@ -8,6 +8,8 @@
 import shutil
 import numpy as np
 import pytest
+import re
+import os
 
 
 ##################### FUNCTIONS USED BY TEST CASES #####################
@@ -1121,21 +1123,33 @@ def get_voltage_data(expt):
 
     # Create RunInfo instance and set scan0 to PropertyScan
     runinfo = ps.RunInfo()
-    runinfo.scan0 = ps.PropertyScan({'v1': ps.drange(0, 0.1, 1)}, prop='voltage', dt=0.000000001)
+    runinfo.scan0 = ps.RepeatScan(1, dt=0.0000001)
 
     # Set RunInfo measure_function (remember, it takes a Experiment object as a parameter and
     # returns an ItemAttribute containing data).
     runinfo.measure_function = get_voltage_data
 
     # Create a Experiment class with the RunInfo and Devices just created
     expt = ps.Experiment(runinfo, devices, time=True)
-    try:
-        expt.run()
-        expt.run()
-        expt.run()
-        expt.run()
-        expt.run()
-        expt.run()
+
+    long_names = []
+
+    while len(long_names) < 3:
         expt.run()
-    except Exception:
-        assert False, f"Fast experiments did not run properly, possible file name overlap. Exception: {Exception}"
+        if len(long_names) == 0:
+            long_names.append(expt.runinfo.long_name)
+        elif (expt.runinfo.long_name[:15] == long_names[0][:15]):
+            long_names.append(expt.runinfo.long_name)
+        else:
+            long_names = [expt.runinfo.long_name]
+
+    err_str = f"First long name '{long_names[0]}' does not match expected date/time format."
+    assert re.match(r'^\d{8}T\d{6}$', long_names[0]), err_str
+    err_str = f"-1 long name '{long_names[1]}' does not match expected increment or format."
+    assert long_names[1] == long_names[0] + '-1', err_str
+    err_str = f"-2 long name '{long_names[1]}' does not match expected increment or format."
+    assert long_names[2] == long_names[0] + '-2', err_str
+
+    for name in long_names:
+        save_path = expt.runinfo.data_path / '{}.hdf5'.format(name)
+        assert os.path.exists(save_path), f"Expected file at path'{save_path}' was not found."