tests: more flexibility in test_attrs.py.

Signed-off-by: Matteo Manica <[email protected]>
rxn4chemistry · Feb 13, 2024 · 3830641 · 3830641
1 parent 8547f6e
commit 3830641
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_attrs.py b/tests/test_attrs.py
@@ -37,4 +37,8 @@ def test_get_variables_and_type_names() -> None:
         ("variable_1", "float"),
         ("variable_2", "str"),
         ("variable_3", "typing.Union[int, NoneType]"),
+    ] or get_variables_and_type_names(DummyClass) == [
+        ("variable_1", "float"),
+        ("variable_2", "str"),
+        ("variable_3", "typing.Optional[int]"),
     ]