def test_laser_singleton(self):
with self.assertRaises(RuntimeError):
class Laser:
results = None
def set_field(self, field_name, value):
self.results = {field_name: value}
laser.set_for_test(Laser())
laser.get()
laser.set_for_test(Laser())
def test_output(self):
drive = laser.get()
time.sleep(10)
drive.turn_off_laser()
data = np.genfromtxt('Data.csv', delimiter=',')
self.assertEqual(data.shape[0], 2)
self.assertTrue(data.shape[1] > 0)
def setUpClass(cls):
cls.drive = laser.get()
def run(self):
laser.get().turn_off_laser()
return
def run_wrapper(self, current_time):
value = self.run(current_time)
laser.get().set_field(self._field_name, value)