Include the python_gut tool

gut_test.py

@@ -0,0 +1,178 @@
+#!/usr/bin/env python3
+
+# Copyright (C) 2015-2016 Andrew Hamilton. All rights reserved.
+# Licensed under the Artistic License 2.0.
+
+import textwrap
+import unittest
+
+import gut
+
+
+class GutTestCase(unittest.TestCase):
+
+    def test_import(self):
+        program = "import hello"
+        self.assertEqual(gut.gut_module(program), program)
+
+    def test_import_and_function(self):
+        program = textwrap.dedent("""
+            import hello
+
+            def first():
+                a = 1
+            """)
+        expected = textwrap.dedent("""
+            import hello
+
+            def first():
+            """)
+        self.assertEqual(gut.gut_module(program), expected)
+
+    def test_import_and_function_and_command(self):
+        program = textwrap.dedent("""
+            import hello
+
+            def first():
+                a = 1
+
+            b = 1
+            """)
+        expected = textwrap.dedent("""
+            import hello
+
+            def first():
+
+            b = 1
+            """)
+        self.assertEqual(gut.gut_module(program), expected)
+
+    def test_import_and_class(self):
+        program = textwrap.dedent("""
+            import hello
+
+            class Foo:
+
+                def bar():
+                    a = 1
+            """)
+        expected = textwrap.dedent("""
+            import hello
+
+            class Foo:
+
+                def bar():
+            """)
+        self.assertEqual(gut.gut_module(program), expected)
+
+    def test_short_blank_line_in_def(self):
+        program = textwrap.dedent("""
+            def bar():
+                a = 1
+
+                b = 2
+            """)
+        expected = textwrap.dedent("""
+            def bar():
+            """)
+        self.assertEqual(gut.gut_module(program), expected)
+
+    def test_nested_functions(self):
+        program = textwrap.dedent("""
+            def bar():
+                a = 1
+                def foo():
+                    pass
+                b = 2
+            """)
+        expected = textwrap.dedent("""
+            def bar():
+            """)
+        self.assertEqual(gut.gut_module(program), expected)
+
+    def test_multiline_signature(self):
+        program = textwrap.dedent("""
+            def bar(a, b
+                    c, d):
+                a = 1
+            """)
+        expected = textwrap.dedent("""
+            def bar(a, b
+                    c, d):
+            """)
+        self.assertEqual(gut.gut_module(program), expected)
+
+    def test_tab_in_indentation(self):
+        program = textwrap.dedent("""
+            def bar():
+                a = 1
+            \tb=2
+            """)
+        expected = textwrap.dedent("""
+            def bar():
+            """)
+        self.assertEqual(gut.gut_module(program), expected)
+
+    def test_comment_in_signature_line(self):
+        program = textwrap.dedent("""
+            def bar(): # comment
+                pass
+            """)
+        expected = textwrap.dedent("""
+            def bar(): # comment
+            """)
+        self.assertEqual(gut.gut_module(program), expected)
+
+    def test_indented_comment_in_body(self):
+        program = textwrap.dedent("""
+            def bar():
+                pass
+                # comment
+                pass
+            """)
+        expected = textwrap.dedent("""
+            def bar():
+            """)
+        self.assertEqual(gut.gut_module(program), expected)
+
+    def test_non_indented_comment_in_body(self):
+        program = textwrap.dedent("""
+            def bar():
+                pass
+            # comment
+                pass
+            """)
+        expected = textwrap.dedent("""
+            def bar():
+
+            # comment
+                pass
+            """)
+        self.assertEqual(gut.gut_module(program), expected)
+
+    def test_non_indented_comment_after_body(self):
+        program = textwrap.dedent("""
+            def bar():
+                pass
+                pass
+            # comment
+            pass
+            """)
+        expected = textwrap.dedent("""
+            def bar():
+
+            # comment
+            pass
+            """)
+        self.assertEqual(gut.gut_module(program), expected)
+
+    def test_commented_out_function(self):
+        program = textwrap.dedent("""
+            # def bar():
+            #     pass
+            """)
+        self.assertEqual(gut.gut_module(program), program)
+
+
+if __name__ == "__main__":
+    unittest.main()