Skip to content

Commit 5711d2d

Browse files
committed
Fix the tests for multiple value hydration.
1 parent b412d50 commit 5711d2d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/test_types.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ def test_can_hydrate_node_structure(self):
170170
struct.append(123)
171171
struct.append(["Person"])
172172
struct.append({"name": "Alice"})
173-
alice = self.value_system.hydrate(struct)
173+
alice, = self.value_system.hydrate([struct])
174174
assert alice.id == 123
175175
assert alice.labels == {"Person"}
176176
assert set(alice.keys()) == {"name"}
@@ -187,7 +187,7 @@ def test_can_hydrate_in_list(self):
187187
struct.append(123)
188188
struct.append(["Person"])
189189
struct.append({"name": "Alice"})
190-
alice_in_list = self.value_system.hydrate([struct])
190+
alice_in_list, = self.value_system.hydrate([[struct]])
191191
assert isinstance(alice_in_list, list)
192192
alice, = alice_in_list
193193
assert alice.id == 123
@@ -200,7 +200,7 @@ def test_can_hydrate_in_dict(self):
200200
struct.append(123)
201201
struct.append(["Person"])
202202
struct.append({"name": "Alice"})
203-
alice_in_dict = self.value_system.hydrate({"foo": struct})
203+
alice_in_dict, = self.value_system.hydrate([{"foo": struct}])
204204
assert isinstance(alice_in_dict, dict)
205205
alice = alice_in_dict["foo"]
206206
assert alice.id == 123

0 commit comments

Comments
 (0)