diff options
| -rw-r--r-- | src/tablib/formats/_yaml.py | 4 | ||||
| -rwxr-xr-x | tests/test_tablib.py | 11 |
2 files changed, 13 insertions, 2 deletions
diff --git a/src/tablib/formats/_yaml.py b/src/tablib/formats/_yaml.py index 408400b..f8f2dff 100644 --- a/src/tablib/formats/_yaml.py +++ b/src/tablib/formats/_yaml.py @@ -13,12 +13,12 @@ class YAMLFormat: def export_set(cls, dataset): """Returns YAML representation of Dataset.""" - return yaml.safe_dump(dataset._package(ordered=False)) + return yaml.safe_dump(dataset._package(ordered=False), default_flow_style=None) @classmethod def export_book(cls, databook): """Returns YAML representation of Databook.""" - return yaml.safe_dump(databook._package(ordered=False)) + return yaml.safe_dump(databook._package(ordered=False), default_flow_style=None) @classmethod def import_set(cls, dset, in_stream): diff --git a/tests/test_tablib.py b/tests/test_tablib.py index 87482e0..5754bb2 100755 --- a/tests/test_tablib.py +++ b/tests/test_tablib.py @@ -1092,6 +1092,17 @@ class YAMLTests(BaseTestCase): self.assertEqual(_yaml, data.yaml) + def test_yaml_export(self): + """YAML export""" + + expected = """\ +- {first_name: John, gpa: 90, last_name: Adams} +- {first_name: George, gpa: 67, last_name: Washington} +- {first_name: Thomas, gpa: 50, last_name: Jefferson} +""" + output = self.founders.yaml + self.assertEqual(output, expected) + class LatexTests(BaseTestCase): def test_latex_export(self): |