class CompanyTests(unittest.TestCase):
def setUp(self):
self.tmdb = TMDb()
self.tmdb.api_key = os.environ["TMDB_API_KEY"]
self.tmdb.language = "en-US"
self.tmdb.debug = True
self.tmdb.wait_on_rate_limit = True
self.tmdb.cache = False
self.company = Company()
def test_get_company_details(self):
details = self.company.details(1)
self.assertTrue(hasattr(details, "description"))
self.assertTrue(hasattr(details, "headquarters"))
self.assertTrue(hasattr(details, "homepage"))
self.assertTrue(hasattr(details, "id"))
self.assertTrue(hasattr(details, "logo_path"))
self.assertTrue(hasattr(details, "name"))
self.assertTrue(hasattr(details, "origin_country"))
self.assertTrue(hasattr(details, "parent_company"))
def test_get_company_alternative_names(self):
alternative_names = self.company.alternative_names(1)
self.assertGreater(len(alternative_names), 0)
for alternative_name in alternative_names:
self.assertTrue(hasattr(alternative_name, "name"))
self.assertTrue(hasattr(alternative_name, "type"))
def test_get_company_images(self):
images = self.company.images(1)
self.assertGreater(len(images), 0)
for image in images:
self.assertTrue(hasattr(image, "aspect_ratio"))
self.assertTrue(hasattr(image, "file_path"))
self.assertTrue(hasattr(image, "height"))
self.assertTrue(hasattr(image, "id"))
self.assertTrue(hasattr(image, "file_type"))
self.assertTrue(hasattr(image, "vote_average"))
self.assertTrue(hasattr(image, "vote_count"))
self.assertTrue(hasattr(image, "width"))
def test_get_company_movies(self):
movies = self.company.movies(1)
self.assertGreater(len(movies), 0)
for movie in movies:
self.assertTrue(hasattr(movie, "id"))
self.assertTrue(hasattr(movie, "title"))
self.assertTrue(hasattr(movie, "overview"))
print(details.id)
print(details.logo_path)
print(details.name)
print(details.origin_country)
print(details.parent_company)
# Company alternative names
alternative_names = company.alternative_names(1)
for alternative_name in alternative_names:
print(alternative_name.name)
print(alternative_name.type)
# Company images
images = company.images(1)
for image in images:
print(image.aspect_ratio)
print(image.file_path)
print(image.height)
print(image.id)
print(image.file_type)
print(image.vote_average)
print(image.vote_count)
print(image.width)
# Company movies
movies = company.movies(1)
for movie in movies:
print(movie.id)