diff --git a/src/sleep.py b/src/sleep.py index 95213ec..bf49342 100644 --- a/src/sleep.py +++ b/src/sleep.py @@ -4,4 +4,7 @@ # that student has slept for def sleep(awake): - pass \ No newline at end of file + tot = 0 + for i in awake: + tot += i + return(tot/len(awake)) \ No newline at end of file diff --git a/testing/sleep_test.py b/testing/sleep_test.py index 32eebd7..a7d4612 100644 --- a/testing/sleep_test.py +++ b/testing/sleep_test.py @@ -8,10 +8,10 @@ class sleepTest(unittest.TestCase): def test1(self): - self.assertEqual(sleep.sleep([5, 23, 15, 17, 19]), 8.2) + self.assertEqual(sleep.sleep([5, 23, 15, 17, 19]), 15.8) def test2(self): - self.assertEqual(sleep.sleep([24, 24, 24, 24, 24, 24, 24]), 0) + self.assertEqual(sleep.sleep([24, 24, 24, 24, 24, 24, 24]), 24) def test3(self): self.assertEqual(sleep.sleep([23, 16, 9, 3, 2, 14, 17]), 12)