def test_calc_average(self):
average_age = utils.calc_average(100, 4, 0)
self.assertEqual(average_age, 25)
average_age = utils.calc_average(100, 3, 1)
self.assertEqual(average_age, 50)
average_age = utils.calc_average(0, 5, 5)
self.assertEqual(average_age, 0)
def test_calc_average(self):
average_age = utils.calc_average(100, 4, 0)
self.assertEqual(average_age, 25)
average_age = utils.calc_average(100, 3, 1)
self.assertEqual(average_age, 50)
average_age = utils.calc_average(0, 5, 5)
self.assertEqual(average_age, 0)
def process_movie(movie, min_accuracy, use_freebase):
cast, crew = tmdb_api.get_movie_credits(movie.id)
none_number = 0;
total_age = 0;
if len(cast) == 0:
#Didn't find any actors in film
return
for person_id in cast:
if person_id in actors_cache:
person = actors_cache[person_id]
else:
person = tmdb_api.get_person_info(person_id)
if not person.birthday and use_freebase:
person.birthday = try_find_birthday_in_freebase(person.name)
actors_cache[person_id] = person
if not person.birthday:
none_number += 1;
else:
total_age += utils.calculate_age(person.birthday)
accuracy = utils.calc_accuracy(len(cast), none_number)
average = utils.calc_average(total_age, len(cast), none_number)
if accuracy >= min_accuracy:
print "'{}' average age is {:.4} with accuracy {:.4}%".format(movie.title.encode('utf-8'), average, accuracy)
def process_movie(movie, min_accuracy, use_freebase):
cast, crew = tmdb_api.get_movie_credits(movie.id)
none_number = 0
total_age = 0
if len(cast) == 0:
#Didn't find any actors in film
return
for person_id in cast:
if person_id in actors_cache:
person = actors_cache[person_id]
else:
person = tmdb_api.get_person_info(person_id)
if not person.birthday and use_freebase:
person.birthday = try_find_birthday_in_freebase(person.name)
actors_cache[person_id] = person
if not person.birthday:
none_number += 1
else:
total_age += utils.calculate_age(person.birthday)
accuracy = utils.calc_accuracy(len(cast), none_number)
average = utils.calc_average(total_age, len(cast), none_number)
if accuracy >= min_accuracy:
print "'{}' average age is {:.4} with accuracy {:.4}%".format(
movie.title.encode('utf-8'), average, accuracy)
