class MovieTest(unittest.TestCase):
    def test_summary(self):
        movie = Movie("X", "1990", "John Doe", Ratings())
        self.assertEqual("TITLE: X YEAR: 1990 DIRECTOR: John Doe AVG RATING: 0.00", movie.summarize())


class RatingsTest(unittest.TestCase):
    def test_ratings(self):
        ratings = Ratings()
        ratings.rate(4)
        ratings.rate(2)
        self.assertEqual(3, ratings.average_rating())

    def test_empty_ratings(self):
        ratings = Ratings()
        self.assertEqual(0, ratings.average_rating())