add tests

pyiron · Jan 28, 2025 · 6f52c9e · 6f52c9e
1 parent 8c99af9
commit 6f52c9e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/semantikon/converter.py b/semantikon/converter.py
@@ -213,4 +213,5 @@ def semantikon_class(cls: type):
             setattr(cls, key, value)  # Append type hints to attributes
     except AttributeError:
         pass
+    cls._is_semantikon_class = True
     return cls
diff --git a/tests/unit/test_dataclass.py b/tests/unit/test_dataclass.py
@@ -26,11 +26,19 @@ def original_price(self):
 @semantikon_class
 @dataclass
 class Output:
-    total_energy: u(float, units="eV", label="TotalEnergy", associate_to_sample=True)
+    total_energy: u(
+        float, units="eV", label="TotalEnergy", associate_to_sample=True
+    )
 
 
 class TestDataclass(unittest.TestCase):
 
+    def test_is_semantikon_class(self):
+        self.assertTrue(hasattr(Pizza, "_is_semantikon_class"))
+        self.assertTrue(Pizza._is_semantikon_class)
+        self.assertTrue(hasattr(Pizza.Topping, "_is_semantikon_class"))
+        self.assertTrue(Pizza.Topping._is_semantikon_class)
+
     def test_type(self):
         self.assertEqual(Pizza.price, int)
         self.assertEqual(Pizza.size, Annotated[float, "centimeter"])