|
|
@ -16,29 +16,38 @@ def read_sample(fname):
|
|
|
|
class MinimizeSource(unittest2.TestCase):
|
|
|
|
class MinimizeSource(unittest2.TestCase):
|
|
|
|
def test_class(self):
|
|
|
|
def test_class(self):
|
|
|
|
original = read_sample('class.py')
|
|
|
|
original = read_sample('class.py')
|
|
|
|
minimized = read_sample('class_min.py')
|
|
|
|
expected = read_sample('class_min.py')
|
|
|
|
self.assertEquals(minimized, minimize_source(original))
|
|
|
|
self.assertEqual(expected, minimize_source(original))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_comment(self):
|
|
|
|
|
|
|
|
original = read_sample('comment.py')
|
|
|
|
|
|
|
|
expected = read_sample('comment_min.py')
|
|
|
|
|
|
|
|
self.assertEqual(expected, minimize_source(original))
|
|
|
|
|
|
|
|
|
|
|
|
def test_def(self):
|
|
|
|
def test_def(self):
|
|
|
|
original = read_sample('def.py')
|
|
|
|
original = read_sample('def.py')
|
|
|
|
minimized = read_sample('def_min.py')
|
|
|
|
expected = read_sample('def_min.py')
|
|
|
|
self.assertEquals(minimized, minimize_source(original))
|
|
|
|
self.assertEqual(expected, minimize_source(original))
|
|
|
|
|
|
|
|
|
|
|
|
def test_hashbang(self):
|
|
|
|
def test_hashbang(self):
|
|
|
|
original = read_sample('hashbang.py')
|
|
|
|
original = read_sample('hashbang.py')
|
|
|
|
minimized = read_sample('hashbang_min.py')
|
|
|
|
expected = read_sample('hashbang_min.py')
|
|
|
|
self.assertEquals(minimized, minimize_source(original))
|
|
|
|
self.assertEqual(expected, minimize_source(original))
|
|
|
|
|
|
|
|
|
|
|
|
def test_mod(self):
|
|
|
|
def test_mod(self):
|
|
|
|
original = read_sample('mod.py')
|
|
|
|
original = read_sample('mod.py')
|
|
|
|
minimized = read_sample('mod_min.py')
|
|
|
|
expected = read_sample('mod_min.py')
|
|
|
|
self.assertEquals(minimized, minimize_source(original))
|
|
|
|
self.assertEqual(expected, minimize_source(original))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_pass(self):
|
|
|
|
|
|
|
|
original = read_sample('pass.py')
|
|
|
|
|
|
|
|
expected = read_sample('pass_min.py')
|
|
|
|
|
|
|
|
self.assertEqual(expected, minimize_source(original))
|
|
|
|
|
|
|
|
|
|
|
|
def test_obstacle_course(self):
|
|
|
|
def test_obstacle_course(self):
|
|
|
|
original = read_sample('obstacle_course.py')
|
|
|
|
original = read_sample('obstacle_course.py')
|
|
|
|
minimized = read_sample('obstacle_course_min.py')
|
|
|
|
expected = read_sample('obstacle_course_min.py')
|
|
|
|
self.assertEquals(minimized, minimize_source(original))
|
|
|
|
self.assertEqual(expected, minimize_source(original))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
if __name__ == '__main__':
|
|
|
|