updates

examples/02105/instructor/week2/stones_tests.py

@@ -6,10 +6,7 @@ from stones import maximum_stones
 # A fancy helper function to generate nicer-looking titles.
 def trlist(x):
     s = str(list(x))
-    if len(s) > 30:
-        s = s[:30] + "...]"
-    return s
-
+    return s[:30] + "...]" if len(s) > 30 else s
 
 class Stones(UTestCase):
     """ Test of the Stones function """
@@ -27,7 +24,7 @@ class Stones(UTestCase):
         self.stest(4, [4]) # One stone weighing 4 kg.
 
     def test_stones2(self):
-        self.stest(4, [1, 4]) # should also give 1
+        self.stest(4, [1, 2, 4]) # should also give 1
 
     def test_stones3(self):
         self.stest(4, [4, 1]) # should also give 1

examples/02105/students/week2/stones_tests.py

@@ -21,13 +21,14 @@ class Stones(UTestCase):
     def test_basecase(self):
         """ Test the stones-example given in the homework """
         N = maximum_stones(15, [2, 5, 3, 1, 8, 4, 5, 7])
+        print("How many stones could we pick up?", N)
         self.assertEqual(N, 5) # Test that we can collect 5 stones.
 
     def test_stones1(self):
         self.stest(4, [4]) # One stone weighing 4 kg.
 
     def test_stones2(self):
-        self.stest(4, [1, 4]) # should also give 1
+        self.stest(4, [1, 2, 4]) # should also give 1
 
     def test_stones3(self):
         self.stest(4, [4, 1]) # should also give 1