def test_get_line_of_data(self):
# Create a device with 1024 simulated pixels
sim = SimulatedUSB()
self.assertRaises(ValueError, sim.get_line_pixel)
self.assertTrue(sim.assign("Stroker785L"))
self.assertEqual(sim.pixel_count, 1024)
pixel_data = sim.get_line_pixel()
self.assertEqual(len(pixel_data), 1024)
self.assertEqual(pixel_data[0], 0)
self.assertEqual(pixel_data[1023], 1023)
# Check a device with 2048 pixels
sim = SimulatedUSB()
self.assertTrue(sim.assign("Stroker785M"))
self.assertEqual(sim.pixel_count, 2048)
pixel_data = sim.get_line_pixel()
self.assertEqual(len(pixel_data), 2048)
self.assertEqual(pixel_data[0], 0)
self.assertEqual(pixel_data[2047], 2047)
def test_simulated_integration_times(self):
# Create a regular simulated device, trigger a long integration,
# expect it to return instantly
sim = SimulatedUSB()
self.assertTrue(sim.assign("Stroker785L"))
self.assertTrue(sim.set_integration_time(3000))
start_time = time.time()
pixel_data = sim.get_line_pixel()
end_time = time.time()
self.assertLess(end_time - start_time, 1)
