@@ -415,23 +415,23 @@ def test_import_google_api_python_client(self):
415
415
416
416
def test_should_return_bigquery_integers_as_python_ints (self ):
417
417
result = gbq ._parse_entry (1 , 'INTEGER' )
418
- tm . assert_equal ( result , int (1 ) )
418
+ assert result == int (1 )
419
419
420
420
def test_should_return_bigquery_floats_as_python_floats (self ):
421
421
result = gbq ._parse_entry (1 , 'FLOAT' )
422
- tm . assert_equal ( result , float (1 ) )
422
+ assert result == float (1 )
423
423
424
424
def test_should_return_bigquery_timestamps_as_numpy_datetime (self ):
425
425
result = gbq ._parse_entry ('0e9' , 'TIMESTAMP' )
426
- tm . assert_equal ( result , np_datetime64_compat ('1970-01-01T00:00:00Z' ) )
426
+ assert result == np_datetime64_compat ('1970-01-01T00:00:00Z' )
427
427
428
428
def test_should_return_bigquery_booleans_as_python_booleans (self ):
429
429
result = gbq ._parse_entry ('false' , 'BOOLEAN' )
430
- tm . assert_equal ( result , False )
430
+ assert result == False
431
431
432
432
def test_should_return_bigquery_strings_as_python_strings (self ):
433
433
result = gbq ._parse_entry ('STRING' , 'STRING' )
434
- tm . assert_equal ( result , 'STRING' )
434
+ assert result == 'STRING'
435
435
436
436
def test_to_gbq_should_fail_if_invalid_table_name_passed (self ):
437
437
with pytest .raises (gbq .NotFoundException ):
@@ -737,7 +737,7 @@ def test_index_column(self):
737
737
private_key = _get_private_key_path ())
738
738
correct_frame = DataFrame (
739
739
{'string_1' : ['a' ], 'string_2' : ['b' ]}).set_index ("string_1" )
740
- tm . assert_equal ( result_frame .index .name , correct_frame .index .name )
740
+ assert result_frame .index .name == correct_frame .index .name
741
741
742
742
def test_column_order (self ):
743
743
query = "SELECT 'a' AS string_1, 'b' AS string_2, 'c' AS string_3"
0 commit comments