@@ -1269,19 +1269,40 @@ def test_to_string_specified_header(self):
1269
1269
df .to_string (header = ['X' ])
1270
1270
1271
1271
def test_to_string_no_index (self ):
1272
- df = DataFrame ({'x' : [1 , 2 , 3 ], 'y' : [4 , 5 , 6 ]})
1273
1272
1274
- df_s = df .to_string (index = False )
1275
- expected = " x y\n 1 4\n 2 5\n 3 6"
1273
+ dfs = [
1276
1274
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 ]}),
1278
1281
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
+ ]
1280
1287
1281
- df_s = df .to_string (index = False )
1282
- expected = " x y\n 1 4\n 2 5\n -3 -6"
1288
+ exs = [
1283
1289
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
1285
1306
1286
1307
def test_to_string_line_width_no_index (self ):
1287
1308
df = DataFrame ({'x' : [1 , 2 , 3 ], 'y' : [4 , 5 , 6 ]})
@@ -1291,10 +1312,17 @@ def test_to_string_line_width_no_index(self):
1291
1312
1292
1313
assert df_s == expected
1293
1314
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 ]})
1295
1323
1296
1324
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 "
1298
1326
1299
1327
assert df_s == expected
1300
1328
0 commit comments