1 files changed, 274 insertions, 0 deletions
diff --git a/tests/test_style.py b/tests/test_style.py
new file mode 100644
index 0000000..206f8a4
--- /dev/null
+++ b/tests/test_style.py
@@ -0,0 +1,274 @@
+import tomli
+
+import tomli_w
+
+
+def test_newline_before_table():
+    actual = tomli_w.dumps({"table": {}})
+    expected = "[table]\n"
+    assert actual == expected
+
+    actual = tomli_w.dumps({"table": {"nested": {}, "val3": 3}, "val2": 2, "val1": 1})
+    expected = """\
+val2 = 2
+val1 = 1
+
+[table]
+val3 = 3
+
+[table.nested]
+"""
+    assert actual == expected
+
+
+def test_empty_doc():
+    assert tomli_w.dumps({}) == ""
+
+
+def test_dont_write_redundant_tables():
+    actual = tomli_w.dumps({"tab1": {"tab2": {"tab3": {}}}})
+    expected = "[tab1.tab2.tab3]\n"
+    assert actual == expected
+
+
+def test_multiline():
+    multiline_string = (
+        "This is longer than threshold!\n"
+        "Should be formatted as a multiline basic string"
+    )
+    actual = tomli_w.dumps({"ml_string": multiline_string}, multiline_strings=True)
+    expected = '''\
+ml_string = """
+This is longer than threshold!
+Should be formatted as a multiline basic string"""
+'''
+    assert actual == expected
+
+
+def test_only_tables():
+    actual = tomli_w.dumps({"tab1": {}, "tab2": {}})
+    expected = """\
+[tab1]
+
+[tab2]
+"""
+    assert actual == expected
+
+
+def test_tricky_keys():
+    actual = tomli_w.dumps({"f": 1, "tab1": {}, "": {"f": 2, "": {"": 1}}, "tab3": {}})
+    expected = """\
+f = 1
+
+[tab1]
+
+[""]
+f = 2
+
+["".""]
+"" = 1
+
+[tab3]
+"""
+    assert actual == expected
+
+
+def test_nested_keys():
+    actual = tomli_w.dumps(
+        {
+            "k": 1,
+            "a": {"b": {"c": {"d": {"e": {"f": {}}, "e2": {"f2": {}}}, "d_key1": 1}}},
+        }
+    )
+    expected = """\
+k = 1
+
+[a.b.c]
+d_key1 = 1
+
+[a.b.c.d.e.f]
+
+[a.b.c.d.e2.f2]
+"""
+    assert actual == expected
+
+
+def test_array_of_tables_containing_lists():
+    example: dict = {"aot": [{"a": [0, 1, 2, 3]}]}
+    expected = """\
+[[aot]]
+a = [
+    0,
+    1,
+    2,
+    3,
+]
+"""
+    actual = tomli_w.dumps(example)
+    assert actual == expected
+    assert tomli.loads(actual) == example
+
+    example = {"a": {"nested": example}}
+    expected = """\
+[[a.nested.aot]]
+a = [
+    0,
+    1,
+    2,
+    3,
+]
+"""
+    actual = tomli_w.dumps(example)
+    assert actual == expected
+
+
+def test_array_of_long_tables():
+    long_dict = {
+        "long-value": "Lorem ipsum sith",
+        "another-long-value": "consectetur adipis",
+        "simple-value": 3,
+    }
+    example = {"table": {"nested-array": [{"a": 42}, long_dict]}}
+    expected = """\
+[[table.nested-array]]
+a = 42
+
+[[table.nested-array]]
+long-value = "Lorem ipsum sith"
+another-long-value = "consectetur adipis"
+simple-value = 3
+"""
+    actual = tomli_w.dumps(example)
+    assert actual == expected
+    assert tomli.loads(actual) == example
+
+
+def test_array_of_short_tables():
+    long_name = "a" * 87
+    example = {"table": {"nested-array": [{long_name: 0}, {"b": 1}, {"c": 2}]}}
+    expected = f"""\
+[table]
+nested-array = [
+    {{ {long_name} = 0 }},
+    {{ b = 1 }},
+    {{ c = 2 }},
+]
+"""
+    actual = tomli_w.dumps(example)
+    assert actual == expected
+
+
+def test_example_issue_12():
+    example = {
+        "table": {
+            "nested_table": [
+                {"array_options": [1, 2, 3]},
+                {"another_array": [1, 2]},
+                {"c": 3},
+            ]
+        }
+    }
+    expected = """\
+[[table.nested_table]]
+array_options = [
+    1,
+    2,
+    3,
+]
+
+[[table.nested_table]]
+another_array = [
+    1,
+    2,
+]
+
+[[table.nested_table]]
+c = 3
+"""
+    actual = tomli_w.dumps(example)
+    assert actual == expected
+    assert tomli.loads(actual) == example
+
+
+def test_table_with_empty_array():
+    # Empty arrays should never be AoTs
+    example: dict = {"table": {"array": []}}
+    expected = """\
+[table]
+array = []
+"""
+    actual = tomli_w.dumps(example)
+    assert actual == expected
+    assert tomli.loads(actual) == example
+
+
+def test_non_trivial_nesting():
+    long = {
+        "long-value": "Lorem ipsum dolor sit amet",
+        "another-long-value": "consectetur adipiscing elit",
+        "a-third-one": "sed do eiusmod tempor incididunt ut labore et dolore magna",
+        "simple-value": 3,
+    }
+    example = {
+        "table": {
+            "aot": [
+                {"nested-table": {"nested_aot": [{"a": [0, 1]}, {"b": 2}, {"c": 3}]}},
+                {"other-nested-table": {"d": 4, "e": 5, "f": [{"g": 6}], "h": [long]}},
+            ]
+        }
+    }
+
+    expected = """\
+[[table.aot]]
+
+[[table.aot.nested-table.nested_aot]]
+a = [
+    0,
+    1,
+]
+
+[[table.aot.nested-table.nested_aot]]
+b = 2
+
+[[table.aot.nested-table.nested_aot]]
+c = 3
+
+[[table.aot]]
+
+[table.aot.other-nested-table]
+d = 4
+e = 5
+f = [
+    { g = 6 },
+]
+
+[[table.aot.other-nested-table.h]]
+long-value = "Lorem ipsum dolor sit amet"
+another-long-value = "consectetur adipiscing elit"
+a-third-one = "sed do eiusmod tempor incididunt ut labore et dolore magna"
+simple-value = 3
+"""
+    actual = tomli_w.dumps(example)
+    assert actual == expected
+    assert tomli.loads(actual) == example
+
+
+def test_multiline_in_aot():
+    data = {"aot": [{"multiline_string": "line1\nline2"}]}
+    assert (
+        tomli_w.dumps(data, multiline_strings=True)
+        == '''\
+[[aot]]
+multiline_string = """
+line1
+line2"""
+'''
+    )
+    assert (
+        tomli_w.dumps(data, multiline_strings=False)
+        == """\
+aot = [
+    { multiline_string = "line1\\nline2" },
+]
+"""
+    )