Skip to content

Commit e90320e

Browse files
author
gshiba
committed
Update/add tests
1 parent abee072 commit e90320e

File tree

1 file changed

+38
-10
lines changed

1 file changed

+38
-10
lines changed

pandas/tests/io/formats/test_format.py

+38-10
Original file line numberDiff line numberDiff line change
@@ -1269,19 +1269,40 @@ def test_to_string_specified_header(self):
12691269
df.to_string(header=['X'])
12701270

12711271
def test_to_string_no_index(self):
1272-
df = DataFrame({'x': [1, 2, 3], 'y': [4, 5, 6]})
12731272

1274-
df_s = df.to_string(index=False)
1275-
expected = " x y\n 1 4\n 2 5\n 3 6"
1273+
dfs = [
12761274

1277-
assert df_s == expected
1275+
# ints
1276+
DataFrame({'x': [1, 2, 3], 'y': [4, 5, 6]}),
1277+
DataFrame({'x': [11, 22, 33], 'y': [4, 5, 6]}),
1278+
DataFrame({'x': [11, 22, -33], 'y': [4, 5, 6]}),
1279+
DataFrame({'x': [11, 22, -33], 'y': [4, 5, -6]}),
1280+
DataFrame({'x': [11, 22, -33], 'y': [44, 55, -66]}),
12781281

1279-
df = DataFrame({'x': [1, 2, -3], 'y': [4, 5, -6]})
1282+
# floats
1283+
DataFrame({'x': [0.1, 0.2, -0.3], 'y': [4, 5, 6]}),
1284+
DataFrame({'x': [0.1, 0.2, -0.3], 'y': [0.4, 0.5, 0.6]}),
1285+
DataFrame({'x': [0.1, 0.2, -0.3], 'y': [0.4, 0.5, -0.6]}),
1286+
]
12801287

1281-
df_s = df.to_string(index=False)
1282-
expected = " x y\n 1 4\n 2 5\n-3 -6"
1288+
exs = [
12831289

1284-
assert df_s == expected
1290+
# ints
1291+
" x y\n 1 4\n 2 5\n 3 6",
1292+
" x y\n 11 4\n 22 5\n 33 6",
1293+
" x y\n 11 4\n 22 5\n-33 6",
1294+
" x y\n 11 4\n 22 5\n-33 -6",
1295+
" x y\n 11 44\n 22 55\n-33 -66",
1296+
1297+
# floats
1298+
" x y\n 0.1 4\n 0.2 5\n-0.3 6",
1299+
" x y\n 0.1 0.4\n 0.2 0.5\n-0.3 0.6",
1300+
" x y\n 0.1 0.4\n 0.2 0.5\n-0.3 -0.6",
1301+
]
1302+
1303+
for df, expected in zip(dfs, exs):
1304+
df_s = df.to_string(index=False)
1305+
assert df_s == expected
12851306

12861307
def test_to_string_line_width_no_index(self):
12871308
df = DataFrame({'x': [1, 2, 3], 'y': [4, 5, 6]})
@@ -1291,10 +1312,17 @@ def test_to_string_line_width_no_index(self):
12911312

12921313
assert df_s == expected
12931314

1294-
df = DataFrame({'x': [1, 2, -3], 'y': [4, 5, -6]})
1315+
df = DataFrame({'x': [11, 22, 33], 'y': [4, 5, 6]})
1316+
1317+
df_s = df.to_string(line_width=1, index=False)
1318+
expected = " x \\\n 11 \n 22 \n 33 \n\n y \n 4 \n 5 \n 6 "
1319+
1320+
assert df_s == expected
1321+
1322+
df = DataFrame({'x': [11, 22, -33], 'y': [4, 5, -6]})
12951323

12961324
df_s = df.to_string(line_width=1, index=False)
1297-
expected = " x \\\n 1 \n 2 \n-3 \n\n y \n 4 \n 5 \n-6 "
1325+
expected = " x \\\n 11 \n 22 \n-33 \n\n y \n 4 \n 5 \n-6 "
12981326

12991327
assert df_s == expected
13001328

0 commit comments

Comments
 (0)