diff --git a/src/ifcopenshell-python/test/test_sql.py b/src/ifcopenshell-python/test/test_sql.py new file mode 100644 index 0000000000..194bc7cf86 --- /dev/null +++ b/src/ifcopenshell-python/test/test_sql.py @@ -0,0 +1,70 @@ +# IfcOpenShell - IFC toolkit and geometry engine +# Copyright (C) 2021 Thomas Krijnen +# +# This file is part of IfcOpenShell. +# +# IfcOpenShell is free software: you can redistribute it and/or modify +# it under the terms of the GNU Lesser General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# IfcOpenShell is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public License +# along with IfcOpenShell. If not, see . + +import pytest +import test.bootstrap +import ifcopenshell +import ifcpatch +from pathlib import Path + +TEST_FILE = Path(__file__).parent / "files" / "basic.ifc" +SQLITE_PATH = None + + +def get_ifc_sqlite() -> ifcopenshell.sqlite: + global SQLITE_PATH + if SQLITE_PATH is None: + SQLITE_PATH = ifcpatch.execute( + {"file": ifcopenshell.open(TEST_FILE), "recipe": "Ifc2Sql", "arguments": ["sqlite"]} + ) + assert isinstance(SQLITE_PATH, str) + ifc_sqlite = ifcopenshell.open(SQLITE_PATH) + assert isinstance(ifc_sqlite, ifcopenshell.sqlite) + return ifc_sqlite + + +class TestFile: + def test_run(self): + ifc_sqlite = get_ifc_sqlite() + assert ifc_sqlite.schema == "IFC4" + + def test_by_id(self): + ifc_sqlite = get_ifc_sqlite() + assert (element := ifc_sqlite.by_id(1)) + assert str(element) == "#1=IFCPROJECT('3kv235yMjDO9tHiTzD8QuS',$,'My Project',$,$,$,$,(#14,#26),#9);" + + def test_by_type(self): + ifc_sqlite = get_ifc_sqlite() + assert (elements := ifc_sqlite.by_type("IfcProject")) + assert str(elements[0]) == "#1=IFCPROJECT('3kv235yMjDO9tHiTzD8QuS',$,'My Project',$,$,$,$,(#14,#26),#9);" + + +class TestEntity: + def test_getattr(self): + ifc_sqlite = get_ifc_sqlite() + assert (element := ifc_sqlite.by_id(1)) + assert element.Name == "My Project" + + def test_setattr(self): + ifc_sqlite = get_ifc_sqlite() + assert (element := ifc_sqlite.by_id(1)) + assert element.Name == "My Project" + element.Name = "Foo" + assert element.Name == "Foo" + element.Name = "My Project" + assert element.Name == "My Project"