Add assessment

test_yahtzee_goals.py

@@ -0,0 +1,25 @@
+from die import Die
+from yahtzee_goals import GoalSmallStraight
+
+def make_dice(faces):
+    """Given a list of face values, returns corresponding dice.
+    This is a helper method used to make testing easier. 
+    """
+    dice = []
+    for face in faces:
+        die = Die()
+        die.face = face
+        dice.append(die)
+    return dice
+
+def test_small_straight(faces, expected):
+    """Tests whether small straight works correctly. 
+    """
+    goal = GoalSmallStraight()
+    result = goal.is_small_straight(make_dice(faces))
+    assert(result == expected)
+
+test_small_straight([1, 1, 2, 3, 4], True)
+test_small_straight([4, 1, 1, 2, 3], True)
+test_small_straight([2, 3, 4, 5, 5], True)
+test_small_straight([1, 3, 3, 4, 5], False)