Codegen: allow to include .py files in schema.py

misc/codegen/test/test_schemaloader.py

@@ -13,7 +13,7 @@ def test_empty_schema():
         pass
 
     assert data.classes == {}
-    assert data.includes == set()
+    assert data.includes == []
     assert data.null is None
     assert data.null_class is None
 
@@ -805,5 +805,51 @@ def test_test_with_double():
                 pass
 
 
+def test_include_dbscheme():
+    @load
+    class data:
+        defs.include("foo.dbscheme")
+        defs.include("bar.dbscheme")
+
+    assert data.includes == ["foo.dbscheme", "bar.dbscheme"]
+
+
+def test_include_source(tmp_path):
+    (tmp_path / "foo.py").write_text("""
+class A(Root):
+    pass
+""")
+    (tmp_path / "bar.py").write_text("""
+class C(Root):
+    pass
+""")
+
+    @load
+    class data:
+        class Root:
+            pass
+
+        defs.include(str(tmp_path / "foo.py"))
+
+        class B(Root):
+            pass
+
+        defs.include(str(tmp_path / "bar.py"))
+
+    assert data.classes == {
+        "Root": schema.Class("Root", derived=set("ABC")),
+        "A": schema.Class("A", bases=["Root"]),
+        "B": schema.Class("B", bases=["Root"]),
+        "C": schema.Class("C", bases=["Root"]),
+    }
+
+
+def test_include_not_supported(tmp_path):
+    with pytest.raises(schema.Error):
+        @load
+        class data:
+            defs.include("foo.bar")
+
+
 if __name__ == '__main__':
     sys.exit(pytest.main([__file__] + sys.argv[1:]))