Updated to pyVHDLModel v0.18.0.

1 files changed, 12 insertions, 4 deletions

diff --git a/testsuite/pyunit/dom/Simple.py b/testsuite/pyunit/dom/Simple.py
index 32033609d..82dd8579e 100644
--- a/testsuite/pyunit/dom/Simple.py
+++ b/testsuite/pyunit/dom/Simple.py
@@ -31,6 +31,7 @@
 # SPDX-License-Identifier: GPL-2.0-or-later
 # ============================================================================
 from pathlib import Path
+from typing import TypeVar, Dict
 from unittest import TestCase
 
 from pyGHDL.dom.NonStandard import Design, Document
@@ -42,6 +43,13 @@ if __name__ == "__main__":
     exit(1)
 
 
+_DictKey = TypeVar("_DictKey")
+_DictValue = TypeVar("_DictValue")
+
+def firstValue(d: Dict[_DictKey, _DictValue]) -> _DictValue:
+    return next(iter(d.values()))
+
+
 class SimpleEntity(TestCase):
     _root = Path(__file__).resolve().parent.parent
     _filename: Path = _root / "dom/examples/SimpleEntity.vhdl"
@@ -71,7 +79,7 @@ class SimpleEntity(TestCase):
 
         self.assertEqual(1, len(design.Documents[0].Entities))
 
-        entity = design.Documents[0].Entities[0]
+        entity = firstValue(design.Documents[0].Entities)
         self.assertEqual("Counter", entity.Identifier)
         print()
         print(entity.Documentation)
@@ -84,7 +92,7 @@ class SimpleEntity(TestCase):
 
         self.assertEqual(1, len(design.Documents[0].Architectures))
 
-        architecture = design.Documents[0].Architectures[0]
+        architecture = firstValue(firstValue(design.Documents[0].Architectures))
         self.assertEqual("rtl", architecture.Identifier)
         print()
         print(architecture.Documentation)
@@ -120,7 +128,7 @@ class SimplePackage(TestCase):
 
         self.assertEqual(1, len(design.Documents[0].Packages))
 
-        package = design.Documents[0].Packages[0]
+        package = firstValue(design.Documents[0].Packages)
         self.assertEqual("utilities", package.Identifier)
         print()
         print(package.Documentation)
@@ -133,7 +141,7 @@ class SimplePackage(TestCase):
 
         self.assertEqual(1, len(design.Documents[0].PackageBodies))
 
-        packageBodies = design.Documents[0].PackageBodies[0]
+        packageBodies = firstValue(design.Documents[0].PackageBodies)
         self.assertEqual("utilities", packageBodies.Identifier)
         print()
         print(packageBodies.Documentation)