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_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_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_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_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_load_shoot_roles(): data = file_load.load_shoot_roles() assert isinstance(data['Ali3n Club F**k']['Workers'], dict)