|
1 | 1 | from csv_diff import load_csv, compare, human_text |
2 | 2 | from .test_csv_diff import ONE, TWO, THREE, FOUR |
| 3 | + |
| 4 | +from textwrap import dedent |
3 | 5 | import io |
4 | 6 |
|
5 | 7 |
|
6 | 8 | def test_row_changed(): |
7 | 9 | diff = compare( |
8 | 10 | load_csv(io.StringIO(ONE), key="id"), load_csv(io.StringIO(TWO), key="id") |
9 | 11 | ) |
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) |
11 | 18 |
|
12 | 19 |
|
13 | 20 | def test_row_added(): |
14 | 21 | diff = compare( |
15 | 22 | load_csv(io.StringIO(THREE), key="id"), load_csv(io.StringIO(TWO), key="id") |
16 | 23 | ) |
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) |
20 | 29 |
|
21 | 30 |
|
22 | 31 | def test_row_removed(): |
23 | 32 | diff = compare( |
24 | 33 | load_csv(io.StringIO(TWO), key="id"), load_csv(io.StringIO(THREE), key="id") |
25 | 34 | ) |
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) |
30 | 40 |
|
31 | 41 |
|
32 | 42 | def test_row_changed_and_row_added(): |
33 | 43 | "Should have headers for each section here" |
34 | 44 | diff = compare( |
35 | 45 | load_csv(io.StringIO(ONE), key="id"), load_csv(io.StringIO(FOUR), key="id") |
36 | 46 | ) |
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