File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -297,9 +297,10 @@ def test_download_dataset_larger_than_200k_rows(self):
297
297
self .assertEqual (len (df .drop_duplicates ()), 200005 )
298
298
299
299
def test_zero_rows (self ):
300
- df = gbq .read_gbq ("SELECT * FROM [publicdata:samples.wikipedia] where timestamp=-9999999" , project_id = PROJECT_ID )
301
- tm .assertIs (df , DataFrame , "Zero rows result should be dataframe." )
302
- self .assertEqual (len (df ), 0 )
300
+ # Bug fix for https://github.com/pydata/pandas/issues/10273
301
+ df = gbq .read_gbq ("SELECT title, language FROM [publicdata:samples.wikipedia] where timestamp=-9999999" , project_id = PROJECT_ID )
302
+ expected_result = DataFrame (columns = ['title' , 'language' ])
303
+ self .assert_frame_equal (df , expected_result )
303
304
304
305
305
306
class TestToGBQIntegration (tm .TestCase ):
You can’t perform that action at this time.
0 commit comments