PEP8 Refactor

Author: Crambor

file_parser.py

@@ -1,6 +1,5 @@
 import os
 import re
-import sys
 
 
 class ParseError(Exception):

unit_tests/file_parsing_tests/unsupported_ext.py

@@ -40,4 +40,3 @@
 DAT
 DAT
 """
-

unit_tests/test_arg_parsing.py

@@ -1,10 +1,9 @@
+from arg_parsing import parse_args
+
 import unittest
 import sys
 import os
 
-sys.path.append(os.path.abspath(""))
-from arg_parsing import parse_args
-
 
 class TestHelp(unittest.TestCase):
     def test_short_form(self):
@@ -292,6 +291,6 @@ def test_exclusive_vq(self):
         self.assertEqual(cm.exception.code, 2)
 
 
+sys.path.append(os.path.abspath(""))
 if __name__ == '__main__':
-    sys.path.append(os.path.abspath(""))
     unittest.main()

unit_tests/test_file_parser.py

@@ -1,11 +1,10 @@
+import file_parser
+
 import unittest
 import json
 import sys
 import os
 
-sys.path.append(os.path.abspath(""))
-import file_parser
-
 
 def reset_directory():
     os.chdir(os.path.dirname(__file__))
@@ -80,3 +79,8 @@ def test_broken_batch_file(self):
         batch_file_path = "unit_tests/file_parsing_tests/broken_batch_test.txt"
         with self.assertRaises(file_parser.ParseError):
             file_parser.parse_batch_test_file(batch_file_path)
+
+
+sys.path.append(os.path.abspath(""))
+if __name__ == '__main__':
+    unittest.main()

unit_tests/test_lmc.py

@@ -1,13 +1,10 @@
-import unittest
+from lmc import LMC
+
 from unittest.mock import patch
+import unittest
 import sys
 import os
 
-sys.path.append(os.path.abspath(""))
-from lmc import LMC
-
-
-
 
 class TestInput(unittest.TestCase):
     def setUp(self):
@@ -132,5 +129,6 @@ def TestExampleProgram(self):
         self.assertTrue(self.lmc.halted)
 
 
+sys.path.append(os.path.abspath(""))
 if __name__ == '__main__':
     unittest.main()

unit_tests/test_lmc_wrapper.py

@@ -1,10 +1,10 @@
+from lmc_wrapper import LMCWrapper
+
 import unittest
 import textwrap
 import sys
 import os
 
-sys.path.append(os.path.abspath(""))
-from lmc_wrapper import LMCWrapper
 
 class TestChecker(unittest.TestCase):
     def setUp(self):
@@ -62,3 +62,8 @@ def setUp(self):
     def test_store_feedback_message(self):
         self.lmc_wrapper.store_feedback_msg([1, 2, 3, 4], [1, 2, 3, 4], 123)
         self.assertEqual(self.lmc_wrapper.feedback, self.correct_feedback)
+
+
+sys.path.append(os.path.abspath(""))
+if __name__ == '__main__':
+    unittest.main()