Merge pull request #23 from mikelolasagasti/fix-TestLoader-addTests

Author: stephenfin

testscenarios/__init__.py

@@ -57,7 +57,7 @@ def test_suite():
     return testscenarios.tests.test_suite()
 
 
-def load_tests(standard_tests, module, loader):
+def load_tests(loader, standard_tests, pattern):
     standard_tests.addTests(loader.loadTestsFromNames(["testscenarios.tests"]))
     return standard_tests
 

testscenarios/tests/__init__.py

@@ -15,20 +15,18 @@
 # limitations under that license.
 
 import doctest
-import sys
 import unittest
 
 import testscenarios
 
 
 def test_suite():
-    standard_tests = unittest.TestSuite()
-    module = sys.modules["testscenarios.tests"]
     loader = unittest.TestLoader()
-    return load_tests(standard_tests, module, loader)
+    standard_tests = unittest.TestSuite()
+    return load_tests(loader, standard_tests, None)
 
 
-def load_tests(standard_tests, module, loader):
+def load_tests(loader, standard_tests, pattern):
     test_modules = [
         "testcase",
         "scenarios",