def test_sign(): aries_json = json.loads('''[{ "name": "Aries", "famous_people": ["Famous1", "Famous2"], "compatibility": ["Leo", "Sagittarius", "Gemini", "Aquarius"], "sun_dates": ["March 21", "April 19"]}]''') signs = get_signs(aries_json) assert (len(signs) == 1) assert (type(signs[0]) == Sign) assert (signs[0].name == 'Aries') assert (signs[0].compatibility == [ 'Leo', 'Sagittarius', 'Gemini', 'Aquarius' ]) assert (signs[0].famous_people == ['Famous1', 'Famous2']) assert (signs[0].sun_dates == ['March 21', 'April 19']) assert (get_signs([]) == [])
def signs(): if not PATH.exists(): urlretrieve(URL, PATH) with open(PATH) as f: data = json.loads(f.read()) return get_signs(data)