Python TubHandler.get_tub_list Examples

Example #1

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

Example #2

def test_get_tub_list(tubs):
    root_dir = tubs[0]
    th = TubHandler(root_dir)
    assert len(th.get_tub_list()) == 5