From bdc0cc7e3ee536f344ed126e153359352aeb001c Mon Sep 17 00:00:00 2001
From: Thomas Jals <jalst31@gmail.com>
Date: Thu, 17 Sep 2026 22:11:11 +0200
Subject: [PATCH] test: assert unwrapped values in elementary_test and unwrap
 tests

---
 tests/test_items.py         | 21 ++++++++++++---------
 tests/test_toml_document.py |  1 +
 tests/util.py               |  7 ++++++-
 3 files changed, 19 insertions(+), 10 deletions(-)

diff --git a/tests/test_items.py b/tests/test_items.py
index f6377d02..33f33dd7 100644
--- a/tests/test_items.py
+++ b/tests/test_items.py
@@ -96,11 +96,11 @@ def test_item_base_has_no_unwrap() -> None:
 
 
 def test_integer_unwrap() -> None:
-    elementary_test(item(666), int)
+    elementary_test(item(666), int, 666)
 
 
 def test_float_unwrap() -> None:
-    elementary_test(item(2.78), float)
+    elementary_test(item(2.78), float, 2.78)
 
 
 @pytest.mark.skipif(
@@ -116,25 +116,25 @@ def test_float_is_not_a_sequence() -> None:
 
 
 def test_false_unwrap() -> None:
-    elementary_test(item(False), bool)
+    elementary_test(item(False), bool, False)
 
 
 def test_true_unwrap() -> None:
-    elementary_test(item(True), bool)
+    elementary_test(item(True), bool, True)
 
 
 def test_datetime_unwrap() -> None:
     dt = datetime.now(tz=timezone.utc)
-    elementary_test(item(dt), datetime)
+    elementary_test(item(dt), datetime, dt)
 
 
 def test_string_unwrap() -> None:
-    elementary_test(item("hello"), str)
+    elementary_test(item("hello"), str, "hello")
 
 
 def test_null_unwrap() -> None:
     n = Null()
-    elementary_test(n, type(None))
+    elementary_test(n, type(None), None)
 
 
 def test_aot_unwrap() -> None:
@@ -147,6 +147,7 @@ def test_aot_unwrap() -> None:
             vu = du[ku]
             assert_is_ppo(ku, str)
             assert_is_ppo(vu, str)
+    assert unwrapped == [{"a": "A"}, {"b": "B"}]
 
 
 def test_aot_set_item() -> None:
@@ -164,12 +165,12 @@ def test_aot_set_item() -> None:
 
 def test_time_unwrap() -> None:
     t = time(3, 8, 14)
-    elementary_test(item(t), time)
+    elementary_test(item(t), time, t)
 
 
 def test_date_unwrap() -> None:
     d = date.today()
-    elementary_test(item(d), date)
+    elementary_test(item(d), date, d)
 
 
 def test_array_unwrap() -> None:
@@ -183,6 +184,7 @@ def test_array_unwrap() -> None:
     assert_is_ppo(a_unwrapped[0], int)
     assert_is_ppo(a_unwrapped[1], float)
     assert_is_ppo(a_unwrapped[2], bool)
+    assert a_unwrapped == [666, 2.78, False]
 
 
 def test_abstract_table_unwrap() -> None:
@@ -197,6 +199,7 @@ def test_abstract_table_unwrap() -> None:
         vu = sub_table[ku]
         assert_is_ppo(ku, str)
         assert_is_ppo(vu, str)
+    assert table_unwrapped == {"table": {"foo": "bar"}, "baz": "borg"}
 
 
 def test_key_comparison() -> None:
diff --git a/tests/test_toml_document.py b/tests/test_toml_document.py
index 7ba61f5e..8479142d 100644
--- a/tests/test_toml_document.py
+++ b/tests/test_toml_document.py
@@ -172,6 +172,7 @@ def test_toml_document_unwrap() -> None:
     assert_is_ppo(unwrapped["tool"], dict)
     assert_is_ppo(next(iter(unwrapped["tool"])), str)
     assert_is_ppo(unwrapped["tool"]["poetry"]["name"], str)
+    assert unwrapped == {"tool": {"poetry": {"name": "foo"}}}
 
 
 def test_toml_document_with_dotted_keys(example: Callable[[str], str]) -> None:
diff --git a/tests/util.py b/tests/util.py
index 83cf5d9f..4e6d5f18 100644
--- a/tests/util.py
+++ b/tests/util.py
@@ -52,6 +52,11 @@ def assert_is_ppo(v_unwrapped: object, unwrapped_type: type) -> None:
     assert isinstance(v_unwrapped, unwrapped_type)
 
 
-def elementary_test(v: Item, unwrapped_type: type) -> None:
+_DEFAULT = object()
+
+
+def elementary_test(v: Item, unwrapped_type: type, expected: object = _DEFAULT) -> None:
     v_unwrapped = v.unwrap()
     assert_is_ppo(v_unwrapped, unwrapped_type)
+    if expected is not _DEFAULT:
+        assert v_unwrapped == expected