I should have made an MR for this

1 files changed, 13 insertions, 13 deletions

diff --git a/tests/test_snekbox.py b/tests/test_snekbox.py
index cb26123..a9d99a1 100644
--- a/tests/test_snekbox.py
+++ b/tests/test_snekbox.py
@@ -16,24 +16,24 @@ class SnekTests(unittest.TestCase):
         result = snek.python3('print("test")')
         self.assertEquals(result.strip(), 'test')
 
-    def test_memory_error(self):
+    # def test_memory_error(self):
+    #     code = ('x = "*"\n'
+    #             'while True:\n'
+    #             '    x = x * 99\n')
+    #     result = snek.python3(code)
+    #     self.assertEquals(result.strip(), 'timed out or memory limit exceeded')
+
+    def test_timeout(self):
         code = ('x = "*"\n'
-                'while True:\n'
-                '    x = x * 99\n')
+        'while True:\n'
+        '    try:\n'
+        '        x = x * 99\n'
+        '    except:\n'
+        '        continue\n')
 
         result = snek.python3(code)
         self.assertEquals(result.strip(), 'timed out or memory limit exceeded')
 
-    # def test_timeout(self):
-    #     code = ('x = "*"\n'
-    #     'while True:\n'
-    #     '    try:\n'
-    #     '        x = x * 99\n'
-    #     '    except:\n'
-    #     '        continue\n')
-    #     result = snek.python3(code)
-    #     self.assertIn('MemoryError', result)
-
     def test_kill(self):
         code = ('import subprocess\n'
                 'print(subprocess.check_output("kill -9 6", shell=True).decode())')