File tree 1 file changed +15
-0
lines changed
1 file changed +15
-0
lines changed Original file line number Diff line number Diff line change @@ -35,6 +35,13 @@ def test1_basic(self):
35
35
# Read full file
36
36
data = read_sas (self .file01 , format = "xport" )
37
37
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 ()
38
45
39
46
# Test incremental read with `read` method.
40
47
reader = read_sas (self .file01 , format = "xport" , iterator = True )
@@ -48,6 +55,14 @@ def test1_basic(self):
48
55
reader .close ()
49
56
tm .assert_frame_equal (data , data_csv .iloc [0 :10 , :])
50
57
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
+
51
66
# Read full file with `read_sas` method
52
67
data = read_sas (self .file01 )
53
68
tm .assert_frame_equal (data , data_csv )
You can’t perform that action at this time.
0 commit comments