Skip to content

Commit 0b43700

Browse files
add test for list/dict
1 parent 9d830b7 commit 0b43700

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

tests/test_output_streams.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -353,6 +353,7 @@ def do_output(self, yaml):
353353

354354
def test_csv_output(self):
355355
yaml = """
356+
- snowfakery_version: 3
356357
- object: foo
357358
fields:
358359
a: 1
@@ -365,6 +366,10 @@ def test_csv_output(self):
365366
fields:
366367
barb: 2
367368
bard: 4
369+
- object: faz
370+
fields:
371+
list_t: ${{fake.json(data_columns={'Spec':'@1.0.1', 'ID':'pyint','Details':{'Name':'name', 'Address':'address'}}, num_rows=2)}}
372+
dict_t: ${{fake.json(data_columns={'Spec':'@1.0.1', 'ID':'pyint','Details':{'Name':'name', 'Address':'address'}}, num_rows=1)}}
368373
"""
369374
with TemporaryDirectory() as t:
370375
output_stream = CSVOutputStream(Path(t) / "csvoutput")
@@ -373,13 +378,15 @@ def test_csv_output(self):
373378
assert messages
374379
assert "foo.csv" in messages[0]
375380
assert "bar.csv" in messages[1]
376-
assert "csvw" in messages[2]
381+
assert "faz.csv" in messages[2]
382+
assert "csvw" in messages[3]
377383
assert (Path(t) / "csvoutput" / "foo.csv").exists()
378384
with open(Path(t) / "csvoutput" / "csvw_metadata.json") as f:
379385
metadata = json.load(f)
380386
assert {table["url"] for table in metadata["tables"]} == {
381387
"foo.csv",
382388
"bar.csv",
389+
"faz.csv",
383390
}
384391

385392
def test_null(self):

0 commit comments

Comments
 (0)