Skip to content

Commit 34c2d29

Browse files
committed
Neater tests using textwrap.dedent
1 parent f74c267 commit 34c2d29

File tree

1 file changed

+30
-16
lines changed

1 file changed

+30
-16
lines changed

tests/test_human_text.py

Lines changed: 30 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,44 +1,58 @@
11
from csv_diff import load_csv, compare, human_text
22
from .test_csv_diff import ONE, TWO, THREE, FOUR
3+
4+
from textwrap import dedent
35
import io
46

57

68
def test_row_changed():
79
diff = compare(
810
load_csv(io.StringIO(ONE), key="id"), load_csv(io.StringIO(TWO), key="id")
911
)
10-
assert '1 row changed\n\n Row 1\n age: "4" => "5"' == human_text(diff)
12+
assert dedent("""
13+
1 row changed
14+
15+
Row 1
16+
age: "4" => "5"
17+
""").strip() == human_text(diff)
1118

1219

1320
def test_row_added():
1421
diff = compare(
1522
load_csv(io.StringIO(THREE), key="id"), load_csv(io.StringIO(TWO), key="id")
1623
)
17-
assert '1 row added\n\n {"id": "2", "name": "Pancakes", "age": "2"}' == human_text(
18-
diff
19-
)
24+
assert dedent("""
25+
1 row added
26+
27+
{"id": "2", "name": "Pancakes", "age": "2"}
28+
""").strip() == human_text(diff)
2029

2130

2231
def test_row_removed():
2332
diff = compare(
2433
load_csv(io.StringIO(TWO), key="id"), load_csv(io.StringIO(THREE), key="id")
2534
)
26-
assert (
27-
'1 row removed\n\n {"id": "2", "name": "Pancakes", "age": "2"}'
28-
== human_text(diff)
29-
)
35+
assert dedent("""
36+
1 row removed
37+
38+
{"id": "2", "name": "Pancakes", "age": "2"}
39+
""").strip() == human_text(diff)
3040

3141

3242
def test_row_changed_and_row_added():
3343
"Should have headers for each section here"
3444
diff = compare(
3545
load_csv(io.StringIO(ONE), key="id"), load_csv(io.StringIO(FOUR), key="id")
3646
)
37-
assert (
38-
"1 row changed, 1 row added\n\n"
39-
"1 row changed\n\n"
40-
" Row 1\n"
41-
' age: "4" => "5"\n\n'
42-
"1 row added\n\n"
43-
' {"id": "3", "name": "Bailey", "age": "1"}'
44-
) == human_text(diff)
47+
assert dedent("""
48+
1 row changed, 1 row added
49+
50+
1 row changed
51+
52+
Row 1
53+
age: "4" => "5"
54+
55+
1 row added
56+
57+
{"id": "3", "name": "Bailey", "age": "1"}
58+
""").strip() == human_text(diff)

0 commit comments

Comments
 (0)