do it properly

Signed-off-by: Chris “Kwpolska” Warrick <kwpolska@gmail.com>
getnikola · Nov 12, 2013 · a2a6201 · a2a6201
1 parent 8d63379
commit a2a6201
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_integration.py b/tests/test_integration.py
@@ -244,14 +244,14 @@ class TestCheck(DemoBuildTest):
     def test_check_links(self):
         with cd(self.target_dir):
             try:
-                main(['check', '-l'])
+                main.main(['check', '-l'])
             except SystemExit as e:
                 self.assertEqual(e.code, 0)
 
     def test_check_files(self):
         with cd(self.target_dir):
             try:
-                main(['check', '-f'])
+                main.main(['check', '-f'])
             except SystemExit as e:
                 self.assertEqual(e.code, 0)
 
@@ -263,7 +263,7 @@ def test_check_links_fail(self):
         with cd(self.target_dir):
             os.unlink(os.path.join("output", "archive.html"))
             try:
-                main(['check', '-l'])
+                main.main(['check', '-l'])
             except SystemExit as e:
                 self.assertNotEqual(e.code, 0)
 
@@ -272,7 +272,7 @@ def test_check_files_fail(self):
             with codecs.open(os.path.join("output", "foobar"), "wb+", "utf8") as outf:
                 outf.write("foo")
             try:
-                main(['check', '-f'])
+                main.main(['check', '-f'])
             except SystemExit as e:
                 self.assertNotEqual(e.code, 0)