diff --git a/src/test/java/edu/bristol/IMDBRatingTest.java b/src/test/java/edu/bristol/IMDBRatingTest.java index 76a9e9c..05121d3 100644 --- a/src/test/java/edu/bristol/IMDBRatingTest.java +++ b/src/test/java/edu/bristol/IMDBRatingTest.java @@ -16,5 +16,11 @@ public void testAverageRating() averageRating = rater.addNewRating(4); assertTrue(averageRating == 3.0, "Adding 2nd rating: average should be 3.0"); + + averageRating = rater.addNewRating(3); + assertTrue(averageRating == 3.0, "Adding 3rd rating: average should be 3.0"); + + averageRating = rater.addNewRating(4); + assertTrue(averageRating == 3.5, "Adding 4th rating: average should be 3.5 CHANGE MADE!!"); } }