add support for plain_data __repr__ with fields that aren't set

[nmutil.git] / src / nmutil / test / test_plain_data.py

diff --git a/src/nmutil/test/test_plain_data.py b/src/nmutil/test/test_plain_data.py
index 6952fe1ff1ca9a52eb14770db4fa8ee4b7406db5..a087e6e5b04f7f05e6f74c911a9dfb9b79b7eea0 100644 (file)
--- a/src/nmutil/test/test_plain_data.py
+++ b/src/nmutil/test/test_plain_data.py
@@ -58,6 +58,15 @@ class PlainDataF2(PlainDataF1):
         self.z = z
 
 
+@plain_data()
+class UnsetField:
+    __slots__ = "a", "b"
+
+    def __init__(self, **kwargs):
+        for name, value in kwargs.items():
+            setattr(self, name, value)
+
+
 class TestPlainData(unittest.TestCase):
     def test_fields(self):
         self.assertEqual(fields(PlainData0), ())
@@ -182,6 +191,11 @@ class TestPlainData(unittest.TestCase):
                          "PlainData2(a=1, b=2, x='x', y='y', z=3)")
         self.assertEqual(repr(PlainDataF2(1, 2, x="x", y="y", z=3)),
                          "PlainDataF2(a=1, b=2, x='x', y='y', z=3)")
+        self.assertEqual(repr(UnsetField()),
+                         "UnsetField(a=<not set>, b=<not set>)")
+        self.assertEqual(repr(UnsetField(a=2)), "UnsetField(a=2, b=<not set>)")
+        self.assertEqual(repr(UnsetField(b=3)), "UnsetField(a=<not set>, b=3)")
+        self.assertEqual(repr(UnsetField(a=5, b=3)), "UnsetField(a=5, b=3)")
 
     def test_frozen(self):
         not_frozen = PlainData0()