Skip to content

Commit ffeff60

Browse files
committed
Add iterator tests for xport
1 parent 00105ee commit ffeff60

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pandas/io/tests/sas/test_xport.py

+15
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,13 @@ def test1_basic(self):
3535
# Read full file
3636
data = read_sas(self.file01, format="xport")
3737
tm.assert_frame_equal(data, data_csv)
38+
num_rows = data.shape[0]
39+
40+
# Test reading beyond end of file
41+
reader = read_sas(self.file01, format="xport", iterator=True)
42+
data = reader.read(num_rows + 100)
43+
self.assertTrue(data.shape[0] == num_rows)
44+
reader.close()
3845

3946
# Test incremental read with `read` method.
4047
reader = read_sas(self.file01, format="xport", iterator=True)
@@ -48,6 +55,14 @@ def test1_basic(self):
4855
reader.close()
4956
tm.assert_frame_equal(data, data_csv.iloc[0:10, :])
5057

58+
# Test read in loop
59+
m = 0
60+
reader = read_sas(self.file01, format="xport", chunksize=100)
61+
for x in reader:
62+
m += x.shape[0]
63+
reader.close()
64+
self.assertTrue(m == num_rows)
65+
5166
# Read full file with `read_sas` method
5267
data = read_sas(self.file01)
5368
tm.assert_frame_equal(data, data_csv)

0 commit comments

Comments
 (0)