@@ -415,23 +415,23 @@ def test_import_google_api_python_client(self):
415415
416416 def test_should_return_bigquery_integers_as_python_ints (self ):
417417 result = gbq ._parse_entry (1 , 'INTEGER' )
418- tm . assert_equal ( result , int (1 ) )
418+ assert result == int (1 )
419419
420420 def test_should_return_bigquery_floats_as_python_floats (self ):
421421 result = gbq ._parse_entry (1 , 'FLOAT' )
422- tm . assert_equal ( result , float (1 ) )
422+ assert result == float (1 )
423423
424424 def test_should_return_bigquery_timestamps_as_numpy_datetime (self ):
425425 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' )
427427
428428 def test_should_return_bigquery_booleans_as_python_booleans (self ):
429429 result = gbq ._parse_entry ('false' , 'BOOLEAN' )
430- tm . assert_equal ( result , False )
430+ assert result == False
431431
432432 def test_should_return_bigquery_strings_as_python_strings (self ):
433433 result = gbq ._parse_entry ('STRING' , 'STRING' )
434- tm . assert_equal ( result , 'STRING' )
434+ assert result == 'STRING'
435435
436436 def test_to_gbq_should_fail_if_invalid_table_name_passed (self ):
437437 with pytest .raises (gbq .NotFoundException ):
@@ -737,7 +737,7 @@ def test_index_column(self):
737737 private_key = _get_private_key_path ())
738738 correct_frame = DataFrame (
739739 {'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
741741
742742 def test_column_order (self ):
743743 query = "SELECT 'a' AS string_1, 'b' AS string_2, 'c' AS string_3"
0 commit comments