diff options
Diffstat (limited to 'testsuite/pyunit/dom')
-rw-r--r-- | testsuite/pyunit/dom/SimpleEntity.py | 8 | ||||
-rw-r--r-- | testsuite/pyunit/dom/__init__.py | 13 |
2 files changed, 20 insertions, 1 deletions
diff --git a/testsuite/pyunit/dom/SimpleEntity.py b/testsuite/pyunit/dom/SimpleEntity.py index 046d5ceff..a8caf9e6f 100644 --- a/testsuite/pyunit/dom/SimpleEntity.py +++ b/testsuite/pyunit/dom/SimpleEntity.py @@ -2,7 +2,6 @@ from pathlib import Path from unittest import TestCase from pyGHDL.dom.Misc import Design, Library, Document -from pyGHDL.dom.DesignUnit import Entity, Architecture if __name__ == "__main__": @@ -37,3 +36,10 @@ class SimpleEntity(TestCase): self.assertEqual(len(design.Documents[0].Entities), 1) self.assertTrue(design.Documents[0].Entities[0].Name == "e1") + def test_Architecture(self): + design = Design() + document = Document(self._path) + design.Documents.append(document) + + self.assertEqual(len(design.Documents[0].Architectures), 1) + self.assertTrue(design.Documents[0].Architectures[0].Name == "behav") diff --git a/testsuite/pyunit/dom/__init__.py b/testsuite/pyunit/dom/__init__.py new file mode 100644 index 000000000..9c103eb6a --- /dev/null +++ b/testsuite/pyunit/dom/__init__.py @@ -0,0 +1,13 @@ +from unittest import TestSuite + +try: + from testsuite.pyunit.dom import SimpleEntity +except ModuleNotFoundError: + from pyunit.dom import SimpleEntity + +def load_tests(loader, testCases, pattern): + suite = TestSuite() + + suite.addTests(loader.loadTestsFromModule(SimpleEntity)) + + return suite |