def setUp(self): self.movieStore = MovieDataStore("test.json") self.movieStore.read() # Note that this assumes the content of test.json # hasn't changed # self.movieTitles = [ \ "How to Train Your Dragon", "Madagascar: Escape 2 Africa", "Megamind", "Madagascar 3: Europe's Most Wanted"]
class MovieTestCase(unittest.TestCase): def setUp(self): self.movieStore = MovieDataStore("test.json") self.movieStore.read() # Note that this assumes the content of test.json # hasn't changed # self.movieTitles = [ \ "How to Train Your Dragon", "Madagascar: Escape 2 Africa", "Megamind", "Madagascar 3: Europe's Most Wanted"] def testMovieStore(self): movies = self.movieStore.movies self.assertEqual(len(movies), 4) for movie in movies: self.assertTrue(movie.title in self.movieTitles)