From 395d8d85b64ee698350affa558216cb6d369b0a0 Mon Sep 17 00:00:00 2001 From: Arjun Chattoraj Date: Wed, 7 Mar 2018 18:12:01 -0500 Subject: [PATCH] fixes #44 --- src/sleep.py | 5 ++++- testing/sleep_test.py | 4 ++-- 2 files changed, 6 insertions(+), 3 deletions(-) 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)