Python: Copy Python extractor to codeql repo

python/extractor/tests/test_import_restrict.py

@@ -0,0 +1,30 @@
+
+import sys
+import os.path
+import shutil
+import unittest
+
+import semmle.populator
+from tests import test_utils
+
+class ExtractorImportRestrictTest(test_utils.ExtractorTest):
+
+    def __init__(self, name):
+      super(ExtractorImportRestrictTest, self).__init__(name)
+      self.module_path = os.path.abspath(os.path.join(self.here, "data-imports"))
+
+    def test_import_unrestricted(self):
+        self.run_extractor("mod1")
+        self.check_only_traps_exists_and_clear("mod1", "mod2", "mod3", "mod4", "mod5")
+
+    def test_import_unrestricted_2(self):
+        self.run_extractor("mod2")
+        self.check_only_traps_exists_and_clear("mod2", "mod3", "mod4", "mod5")
+
+    def test_import_depth(self):
+        self.run_extractor("--max-import-depth", "1", "mod1")
+        self.check_only_traps_exists_and_clear("mod1", "mod2")
+
+    def test_import_depth_2(self):
+        self.run_extractor("--max-import-depth", "2", "mod1")
+        self.check_only_traps_exists_and_clear("mod1", "mod2", "mod3", "mod4")