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)