def test_year(self): db = Database() db.store_actor("actor_test1.txt") db.store_movie("movie_test1.txt") graph = Graph(db) self.assertEqual(graph.actor_in_year(1995),"abc") self.assertEqual(graph.movie_in_year(1995),"movie1,movie2,movie3")
def test_store(self): db = Database() db.store_actor("actor_test1.txt") db.store_movie("movie_test1.txt") graph = Graph(db) self.assertEqual(len(graph.actor_dict), 3) self.assertEqual(len(graph.movie_dict), 2)
def test_old(self): db = Database() db.store_actor("actor_test1.txt") db.store_movie("movie_test1.txt") graph = Graph(db) self.assertEqual(graph.get_old(1), "ddd") self.assertEqual(graph.get_old(2), "ddd,abc")
def test_gross(self): db = Database() db.store_actor("actor_test1.txt") db.store_movie("movie_test1.txt") graph = Graph(db) self.assertEqual(graph.find_gross("abc"), 111000000000) self.assertEqual(graph.find_gross("Garden of Evil"), 3100000 )
def test_db_value(self): db = Database() db.store_actor("actor_test1.txt") db.store_movie("movie_test1.txt") actor = db.actor[0] self.assertEqual(actor.name, "abc") self.assertEqual(actor.age," 23") self.assertEqual(actor.age_int,23)
def test_type(self): db = Database() db.store_actor("actor_test1.txt") db.store_movie("movie_test1.txt") graph = Graph(db) dict = {} self.assertEqual(type(graph.actor_dict), type(dict)) self.assertEqual(type(graph.movie_dict), type(dict))
def test_db_actor_movie(self): db = Database() db.store_actor("actor_test1.txt") db.store_movie("movie_test1.txt") actor = db.actor[0] print(actor.movie[0]) self.assertEqual(actor.movie[0], "movie1") self.assertEqual(actor.movie[1], "movie2") self.assertEqual(actor.movie[2], "movie3")
return data def query(self): # 1. Find how much a movie has grossed print(1, self.find_gross("Garden of Evil")) # 2. Find which movies an actor has worked in print(2, self.search_movie_by_actor("Marisa Tomei")) # 3. List which actors worked in a movie print(3, self.search_actor_by_movie("Garden of Evil")) # 4. List the top X actors with the most total grossing value print(4, self.get_top(3)) # List the oldest X actors print(5, self.get_old(3)) # List all the movies for a given year print(6, self.actor_in_year(1972)) # List all the actors for a given year print(7, self.movie_in_year(1977)) db = Database() db.store_actor("actor.txt") db.store_movie("movie.txt") graph = Graph(db) graph.query()
def test_db_count(self): db = Database() db.store_movie("movie.txt") db.store_actor("actor.txt") self.assertTrue(len(db.actor)>250) self.assertTrue(len(db.movie) > 125)
def test_db_store(self): db = Database() db.store_actor("actor_test1.txt") db.store_movie("movie_test1.txt") self.assertEqual(len(db.actor),3) self.assertEqual(len(db.movie),2)