def test_high_fps_system(self): logger = main_cam(phone_cam=False, video_path="tests/images/system_test_120fps.mp4", default_pen_color=[174, 190, 173]) # logger.save_to_disk("tests/system/test_high_fps_info.csv") right = pd.read_csv("tests/system/test_high_fps_info.csv") pd.testing.assert_frame_equal(logger.memory, right)
def test_simple_system(self): logger = main_cam( phone_cam=False, video_path="tests/images/simple_system_test_30fps.mp4", default_pen_color=[178, 230, 80]) # logger.save_to_disk("tests/system/simple_system_info.csv") right = pd.read_csv("tests/system/simple_system_info.csv") pd.testing.assert_frame_equal(logger.memory, right)
def test_simple_system(self): logger = main_cam(phone_cam=False, video_path="tests/images/load_test_30fps.mp4", default_pen_color=[0, 0, 255]) procs = cpu_count() pool = Pool(procs) pool.map(f, range(procs * 4)) exit_child(0, 0) right = pd.read_csv("tests/system/load_system_info.csv") pd.testing.assert_frame_equal(logger.memory, right)
def test_simple_system(self): imgs_paths = [ "tests/images/hand_0.jpeg", "tests/images/hand_1.jpeg", None, None, "tests/images/hand_2.jpeg", "tests/images/hand_3.jpeg" ] logger = main_cam(phone_cam=False, imgs_paths=imgs_paths, default_pen_color=[178, 230, 80]) # logger.save_to_disk("tests/system/recovery_info.csv") right = pd.read_csv("tests/system/recovery_info.csv") pd.testing.assert_frame_equal(logger.memory, right)
from mp.main_cam_loop import main_cam if __name__ == '__main__': phone_cam = True main_cam(phone_cam)