def test_new_tub_writer(tubs): root_dir = tubs[0] th = TubHandler(root_dir) inputs=['cam/image_array', 'angle', 'throttle'] types=['image_array', 'float', 'float'] tw = th.new_tub_writer(inputs, types) assert len(th.get_tub_list()) == 6 print(tw.path) assert int(tw.path.split('_')[-2]) == 5
def test_get_tub_list(tubs): root_dir = tubs[0] th = TubHandler(root_dir) assert len(th.get_tub_list()) == 5