fixed a bug in an old schema test

Author: Vedant1

dsi/core.py

@@ -484,6 +484,8 @@ def artifact_handler(self, interaction_type, query = None, **kwargs):
                     if self.debug_level != 0:
                         self.logger.error(f"Error ingesting data in {original_file} @ line {return_line_number} due to {errorMessage[1]}")
                     if self.user_wrapper:
+                        if errorMessage[1].startswith("A complex schema"):
+                            raise errorMessage[0](errorMessage[1])
                         raise errorMessage[0](f"Error ingesting data due to {errorMessage[1]}")
                     else:
                         raise errorMessage[0](f"Error ingesting data in {original_file} @ line {return_line_number} due to {errorMessage[1]}")

dsi/dsi.py

@@ -100,7 +100,7 @@ def schema(self, filename = None):
             if not os.path.exists(filename):
                 sys.exit("schema() ERROR: Input schema file must have a valid filepath. Please check again.")
             if "dsi_relations" in self.t.active_metadata:
-                sys.exit("schema() ERROR: Input schema file must have a valid filepath. Please check again.")
+                sys.exit("schema() ERROR: There is already a complex schema in memory. First load all its associated files.")
 
             fnull = open(os.devnull, 'w')
             with redirect_stdout(fnull):

dsi/tests/test_dsi.py

@@ -858,10 +858,9 @@ def test_error_schema_sqlite_backend():
 
     try:
         test.schema(filename="examples/test/yaml1_schema.json")
-        test.query("SELECT * FROM math") # Querying data but need to load in associated data after loading in schema
         assert False
     except SystemExit as e:
-        expected = "ERROR: Cannot query() on an empty backend. Please ensure there is data in it."
+        expected = "schema() ERROR: There is already a complex schema in memory. First load all its associated files."
         assert str(e) == expected
 
 def test_query_update_schema_sqlite_backend():
@@ -1797,5 +1796,6 @@ def test_fail_overwrite_schema_duckdb_backend():
     try:
         test.schema(filename="examples/test/yaml1_circular_schema.json") # should not allow circular dependency overwrite
         assert False
-    except:
-        assert True
+    except SystemExit as e:
+        expected = "schema() ERROR: A complex schema with a circular dependency cannot be ingested into a DuckDB backend."
+        assert str(e) == expected