def test_stimulator(StimulatorClass,
InterfaceClass,
remove_db_file=True,
*args,
**kwargs):
tmp = tempfile.mkstemp(suffix="_ethoscope_test.db")[1]
print("Making a tmp db: " + tmp)
cam = MovieVirtualCamera(VIDEO, drop_each=15)
rb = SleepMonitorWithTargetROIBuilder()
rois = rb.build(cam)
cam.restart()
connection = HardwareConnection(InterfaceClass)
try:
# stimulators = [MockSDStimulator(connection,min_inactive_time= 10) for _ in rois ]
stimulators = [
StimulatorClass(connection, *args, **kwargs) for _ in rois
]
mon = Monitor(cam, AdaptiveBGModel, rois, stimulators=stimulators)
drawer = DefaultDrawer(draw_frames=DRAW_FRAMES)
with SQLiteResultWriter(tmp, rois) as rw:
mon.run(result_writer=rw, drawer=drawer)
# cred = {"name": "ethoscope_db",
# "user": "******",
# "password": "******"}
# with ResultWriter( cred , rois) as rw:
# mon.run(result_writer=rw, drawer=drawer)
finally:
if remove_db_file:
print("Removing temp db (" + tmp + ")")
os.remove(tmp)
else:
print("db file lives in (" + tmp + ")")
connection.stop()
def test_API(self):
random.seed(1)
cam = MovieVirtualCamera(VIDEO)
rb = SleepMonitorWithTargetROIBuilder()
rois = rb.build(cam)
hc = HardwareConnection(MockInterface)
stimulators = [MockStimulator(hc) for _ in rois]
cam.restart()
mon = Monitor(cam, AdaptiveBGModel, rois, stimulators)
drawer = DefaultDrawer(draw_frames=DRAW_FRAMES)
tmp = tempfile.mkstemp(suffix="_ethoscope_test.db")[1]
try:
print("Making a tmp db: " + tmp)
with SQLiteResultWriter(tmp, rois) as rw:
mon.run(result_writer=rw, drawer=drawer)
except:
self.fail("testAPI raised ExceptionType unexpectedly!")
finally:
hc.stop()
cam._close()
print("Removing temp db (" + tmp + ")")
os.remove(tmp)
class MockSDInterface(MockLynxMotionInterface, SleepDepriverInterface):
pass
class MockSDStimulator(SleepDepStimulator):
_HardwareInterfaceClass = MockSDInterface
tmp = tempfile.mkstemp(suffix="_ethoscope_test.db")[1]
print("Making a tmp db: " + tmp)
cam = MovieVirtualCamera(VIDEO, drop_each=15)
rb = SleepMonitorWithTargetROIBuilder()
rois = rb.build(cam)
cam.restart()
connection = HardwareConnection(MockSDInterface)
stimulators = [
MockSDStimulator(connection, min_inactive_time=10) for _ in rois
]
mon = Monitor(cam, AdaptiveBGModel, rois, stimulators=stimulators)
drawer = DefaultDrawer(draw_frames=DRAW_FRAMES)
try:
with SQLiteResultWriter(tmp, rois) as rw:
mon.run(result_writer=rw, drawer=drawer)
finally:
print("Removing temp db (" + tmp + ")")
os.remove(tmp)
connection.stop()