def test_get_license_with_missing_license_id(): license_info = { '1': ('by-nc-sa', '2.0'), '2': ('by-nc', '2.0'), } actual_license, actual_license_version = flickr._get_license( 12, license_info=license_info) assert actual_license is None assert actual_license_version is None
def test_get_license_with_str_license_id(): license_info = { '1': ('by-nc-sa', '2.0'), '2': ('by-nc', '2.0'), } actual_license, actual_license_version = flickr._get_license( '2', license_info=license_info) expect_license, expect_license_version = 'by-nc', '2.0' assert expect_license == actual_license assert expect_license_version == actual_license_version