Issue #160: Add minimize_source testcases

test.sh

@@ -46,6 +46,7 @@ run_test tests/importer_test.py
 run_test tests/latch_test.py
 run_test tests/local_test.py
 run_test tests/master_test.py
+run_test tests/minimize_source_tests.py
 run_test tests/module_finder_test.py
 run_test tests/nested_test.py
 run_test tests/parent_test.py

tests/data/minimize_samples/class.py

@@ -0,0 +1,8 @@
+class C:
+    """docstring
+    """
+
+    def method(self):
+        """docstring
+        """
+        pass

tests/data/minimize_samples/class_min.py

@@ -0,0 +1,8 @@
+class C:
+
+
+
+    def method(self):
+
+
+        pass

tests/data/minimize_samples/comment.py

@@ -0,0 +1,2 @@
+# comment
+    # comment

tests/data/minimize_samples/comment_min.py

@@ -0,0 +1 @@
+

tests/data/minimize_samples/def.py

@@ -0,0 +1,4 @@
+def f():
+    """docstring
+    """
+    pass

tests/data/minimize_samples/def_min.py

@@ -0,0 +1,4 @@
+def f():
+
+
+    pass

tests/data/minimize_samples/hashbang.py

@@ -0,0 +1,3 @@
+#/usr/bin/python -c
+# coding: utf-8
+# comment

tests/data/minimize_samples/hashbang_min.py

@@ -0,0 +1,3 @@
+#/usr/bin/python -c
+# coding: utf-8
+

tests/data/minimize_samples/mod.py

@@ -0,0 +1,5 @@
+
+"""docstring
+"""
+
+pass

tests/data/minimize_samples/mod_min.py

@@ -0,0 +1,5 @@
+
+
+
+
+pass

tests/data/minimize_samples/obstacle_course.py

@@ -0,0 +1,49 @@
+#!python
+# coding : utf-8
+# comment
+"docstring"
+
+import sys
+
+"cc", {}
+
+def f1(a, b= None, c=[]): # comment
+ r"""docstring
+ """
+ x = ""		
+ print(
+ a,  
+    
+b,
+ """arg1""", # comment
+"""arg2""", # comment
+ c )  
+ # comment
+ 'foo' > sys.stderr
+ "baz"
+#comment
+ sys.stdout;lambda: "justastring"
+
+1j
+
+def f2():
+    # comment
+    '''docstring'''
+    pass
+class c:
+    
+    u'''docstring'''
+
+    
+    f='justastring'
+    b'''docstring
+'''
+@f1
+class c2(object):
+    'docstring'
+    def __init__(self):
+        "docstring"
+        def inner(): pass
+        d = {'a':0}
+        for x in '':
+            "justastring"

tests/data/minimize_samples/obstacle_course_min.py

@@ -0,0 +1,49 @@
+#!python
+# coding : utf-8
+
+
+
+import sys
+
+"cc", {}
+
+def f1(a, b= None, c=[]):
+
+
+ x = ""
+ print(
+ a,
+
+b,
+ """arg1""",
+"""arg2""",
+ c )
+
+ 'foo' > sys.stderr
+
+
+ sys.stdout;lambda: "justastring"
+
+1j
+
+def f2():
+
+
+    pass
+class c:
+
+
+
+
+    f='justastring'
+
+
+@f1
+class c2(object):
+
+    def __init__(self):
+
+        def inner(): pass
+        d = {'a':0}
+        for x in '':
+            "justastring"

tests/data/minimize_samples/pass.py

@@ -0,0 +1 @@
+pass

tests/data/minimize_samples/pass_min.py

@@ -0,0 +1 @@
+pass

tests/minimize_source_test.py

@@ -0,0 +1,45 @@
+import unittest2
+
+from mitogen.parent import minimize_source
+
+import testlib
+
+
+def read_sample(fname):
+    sample_path = testlib.data_path('minimize_samples/' + fname)
+    sample_file = open(sample_path)
+    sample = sample_file.read()
+    sample_file.close()
+    return sample
+
+
+class MinimizeSource(unittest2.TestCase):
+    def test_class(self):
+        original = read_sample('class.py')
+        minimized = read_sample('class_min.py')
+        self.assertEquals(minimized, minimize_source(original))
+
+
+    def test_def(self):
+        original = read_sample('def.py')
+        minimized = read_sample('def_min.py')
+        self.assertEquals(minimized, minimize_source(original))
+
+    def test_hashbang(self):
+        original = read_sample('hashbang.py')
+        minimized = read_sample('hashbang_min.py')
+        self.assertEquals(minimized, minimize_source(original))
+
+    def test_mod(self):
+        original = read_sample('mod.py')
+        minimized = read_sample('mod_min.py')
+        self.assertEquals(minimized, minimize_source(original))
+
+    def test_obstacle_course(self):
+        original = read_sample('obstacle_course.py')
+        minimized = read_sample('obstacle_course_min.py')
+        self.assertEquals(minimized, minimize_source(original))
+
+
+if __name__ == '__main__':
+    unittest2.main()