Format with Black

kazk · kazk · commit efccb9292874 · 2020-07-12T17:41:09.000-07:00
https://github.com/psf/black
diff --git a/codewars_test/test_framework.py b/codewars_test/test_framework.py
@@ -10,16 +10,19 @@ def format_message(message):
 
 
 def display(type, message, label="", mode=""):
-    print("\n<{0}:{1}:{2}>{3}".format(
-        type.upper(), mode.upper(), label, format_message(message)))
+    print(
+        "\n<{0}:{1}:{2}>{3}".format(
+            type.upper(), mode.upper(), label, format_message(message)
+        )
+    )
 
 
 def expect(passed=None, message=None, allow_raise=False):
     if passed:
-        display('PASSED', 'Test Passed')
+        display("PASSED", "Test Passed")
     else:
         message = message or "Value is not what was expected"
-        display('FAILED', message)
+        display("FAILED", message)
         if allow_raise:
             raise AssertException(message)
 
@@ -67,14 +70,17 @@ def expect_no_error(message, function, exception=BaseException):
     pass_()
 
 
-def pass_(): expect(True)
+def pass_():
+    expect(True)
 
 
-def fail(message): expect(False, message)
+def fail(message):
+    expect(False, message)
 
 
 def assert_approx_equals(
-        actual, expected, margin=1e-9, message=None, allow_raise=False):
+    actual, expected, margin=1e-9, message=None, allow_raise=False
+):
     msg = "{0} should be close to {1} with absolute or relative margin of {2}"
     equals_msg = msg.format(repr(actual), repr(expected), repr(margin))
     if message is None:
@@ -85,14 +91,14 @@ def assert_approx_equals(
     expect(abs((actual - expected) / div) < margin, message, allow_raise)
 
 
-'''
+"""
 Usage:
 @describe('describe text')
 def describe1():
     @it('it text')
     def it1():
         # some test cases...
-'''
+"""
 
 
 def _timed_block_factory(opening_text):
@@ -110,44 +116,49 @@ def wrapper(func):
             try:
                 func()
             except AssertionError as e:
-                display('FAILED', str(e))
+                display("FAILED", str(e))
             except Exception:
-                fail('Unexpected exception raised')
-                tb_str = ''.join(format_exception(*exc_info()))
-                display('ERROR', tb_str)
-            display('COMPLETEDIN', '{:.2f}'.format((timer() - time) * 1000))
+                fail("Unexpected exception raised")
+                tb_str = "".join(format_exception(*exc_info()))
+                display("ERROR", tb_str)
+            display("COMPLETEDIN", "{:.2f}".format((timer() - time) * 1000))
             if callable(after):
                 after()
+
         return wrapper
+
     return _timed_block_decorator
 
 
-describe = _timed_block_factory('DESCRIBE')
-it = _timed_block_factory('IT')
+describe = _timed_block_factory("DESCRIBE")
+it = _timed_block_factory("IT")
 
 
-'''
+"""
 Timeout utility
 Usage:
 @timeout(sec)
 def some_tests():
     any code block...
 Note: Timeout value can be a float.
-'''
+"""
 
 
 def timeout(sec):
     def wrapper(func):
         from multiprocessing import Process
-        msg = 'Should not throw any exceptions inside timeout'
+
+        msg = "Should not throw any exceptions inside timeout"
 
         def wrapped():
             expect_no_error(msg, func)
+
         process = Process(target=wrapped)
         process.start()
         process.join(sec)
         if process.is_alive():
-            fail('Exceeded time limit of {:.3f} seconds'.format(sec))
+            fail("Exceeded time limit of {:.3f} seconds".format(sec))
             process.terminate()
             process.join()
+
     return wrapper
