def test_video_add_image(): video = list(file_load.get_table().items())[0][1] images = file_load.get_images() assert video.get('image') is None video = utils.video_add_images(video, images) assert video.get('image') is not None
def test_video_add_worker_and_roles(): shoot_roles = file_load.load_shoot_roles() video = list(file_load.get_table().items())[0][1] assert video.get('Workers') is None video = utils.video_add_worker_and_roles(video, shoot_roles) assert isinstance(video['Workers'], dict)
def test_image_from_video_url(): table = file_load.get_table() for name, video in table.items(): stripped_name = name.replace(',','').replace(' ','') vcr_path = 'test/vcr/test_IWC_integration/video{}.yaml'.format(stripped_name) with vcr.use_cassette('test/vcr/test_IWC_integration/video_{}.yaml'.format(stripped_name)): image = IWC_integration.image_from_video_url(video['link']) assert image[-4:] == '.gif'
def test_video_add_role_unscaled_percents(): shoot_roles = file_load.load_shoot_roles() role_percents = file_load.load_role_percents() video = list(file_load.get_table().items())[0][1] video = utils.video_add_worker_and_roles(video, shoot_roles) assert video.get('role percents unscaled') is None video = utils.video_add_role_unscaled_percents(video, role_percents) assert 0 <= video['role percents unscaled']['QAPC'] <= 100
def test_video_create_scaling_factor(): shoot_roles = file_load.load_shoot_roles() role_percents = file_load.load_role_percents() video = list(file_load.get_table().items())[0][1] video = utils.video_add_worker_and_roles(video, shoot_roles) video = utils.video_add_role_unscaled_percents(video, role_percents) assert video.get('scaling factor') is None video = utils.video_create_scaling_factor(video) assert 0 <= video.get('scaling factor') <= 1
def test_video_get_worker_earnings(): shoot_roles = file_load.load_shoot_roles() role_percents = file_load.load_role_percents() video = list(file_load.get_table().items())[0][1] video = utils.video_add_worker_and_roles(video, shoot_roles) video = utils.video_add_role_unscaled_percents(video, role_percents) video = utils.video_create_scaling_factor(video) video = utils.video_scale_role_percents(video) video = utils.video_get_total_earnings(video) assert video.get('earnings') is None video = utils.video_get_worker_earnings(video) assert isinstance(video.get('earnings'), dict)
def test_scaling_factor_applies_properly(): shoot_roles = file_load.load_shoot_roles() role_percents = file_load.load_role_percents() video = list(file_load.get_table().items())[0][1] video = utils.video_add_worker_and_roles(video, shoot_roles) video = utils.video_add_role_unscaled_percents(video, role_percents) video = utils.video_create_scaling_factor(video) video = utils.video_scale_role_percents(video) expected_scaled_percent = video['role percents unscaled']['QAPC'] * video['scaling factor'] scaled_percent = video['role percents']['QAPC'] assert expected_scaled_percent == scaled_percent
def test_scaling_factor_applies_properly(): shoot_roles = file_load.load_shoot_roles() role_percents = file_load.load_role_percents() video = list(file_load.get_table().items())[0][1] video = utils.video_add_worker_and_roles(video, shoot_roles) video = utils.video_add_role_unscaled_percents(video, role_percents) video = utils.video_create_scaling_factor(video) video = utils.video_scale_role_percents(video) expected_scaled_percent = video['role percents unscaled']['QAPC'] * video[ 'scaling factor'] scaled_percent = video['role percents']['QAPC'] assert expected_scaled_percent == scaled_percent
def test_all_videos(): shoot_roles = file_load.load_shoot_roles() role_percents = file_load.load_role_percents() for video in file_load.get_table().values(): video = utils.video_add_worker_and_roles(video, shoot_roles) video = utils.video_add_role_unscaled_percents(video, role_percents) video = utils.video_create_scaling_factor(video) video = utils.video_scale_role_percents(video) video = utils.video_get_total_earnings(video) video = utils.video_get_worker_earnings(video) total_earnings = video['total earnings'] sum_all_earnings = 0 for earning in video['earnings'].values(): sum_all_earnings += earning assert round(total_earnings, 2) == round(sum_all_earnings, 2)
def get_and_populate_shoot_table(): shoot_roles = file_load.load_shoot_roles() role_percents = file_load.load_role_percents() images = file_load.get_images() table = file_load.get_table() for name, video in table.items(): video = utils.video_add_worker_and_roles(video, shoot_roles) video = utils.video_add_role_unscaled_percents(video, role_percents) video = utils.video_create_scaling_factor(video) video = utils.video_scale_role_percents(video) video = utils.video_get_total_earnings(video) video = utils.video_get_worker_earnings(video) video = utils.video_add_images(video, images) table[name] = video return table
def test_get_table(): data = file_load.get_table() assert isinstance(data, collections.OrderedDict) assert bool(data)
def test_get_first_video_from_table(): video = list(file_load.get_table().items())[0][1] assert isinstance(video, dict) assert bool(video)