closes #176

Author: sofiiako

Loops/Fix infinite execution/task-info.yaml

@@ -14,3 +14,5 @@ files:
   visible: false
 - name: stop_run.png
   visible: false
+- name: tests/decorated_test_function.py
+  visible: false

Loops/Fix infinite execution/tests/decorated_test_function.py

@@ -0,0 +1,22 @@
+import wrapt_timeout_decorator
+import contextlib
+import io
+
+from infinite import should_not_be_infinite
+
+timeoutlimit = 5
+
+
+@wrapt_timeout_decorator.timeout(timeoutlimit)
+def test_that_can_take_too_long():
+    f = io.StringIO()
+    with contextlib.redirect_stdout(f):
+        should_not_be_infinite()
+    output = f.getvalue().split('\n')
+    try:
+        if ['Hello, World!', 'Hello, World!', 'Hello, World!', 'Hello, World!', 'Hello, World!', ''] == output:
+            return
+        else:
+            raise AssertionError
+    except AssertionError as e:
+        return str(e)

Loops/Fix infinite execution/tests/test_task.py

@@ -1,20 +1,16 @@
 import unittest
-import contextlib
-import io
+from tests.decorated_test_function import test_that_can_take_too_long
 
-import wrapt_timeout_decorator
-
-from infinite import should_not_be_infinite
-
-
-class TestCaseWithTimeouts(unittest.TestCase):
-    @wrapt_timeout_decorator.timeout(2)
-    def test_that_can_take_too_long(self):
-        f = io.StringIO()
-        with contextlib.redirect_stdout(f):
-            should_not_be_infinite()
-        output = f.getvalue().split('\n')
-
-        self.assertEqual(['Hello, World!', 'Hello, World!', 'Hello, World!', 'Hello, World!', 'Hello, World!', ''], output)
+timeoutlimit = 5
 
 
+class TestCase(unittest.TestCase):
+    def test_infinite(self):
+        try:
+            err = test_that_can_take_too_long()
+            if err:
+                self.fail(msg=err)
+        except TimeoutError as e:
+            self.fail(
+                msg=f"TimeoutError after {timeoutlimit} seconds. Your method's execution does not seem to end in a "
+                    "reasonable amount of time.")