|
9 | 9 | import logging
|
10 | 10 |
|
11 | 11 | import numpy as np
|
12 |
| -import pytest |
13 | 12 |
|
14 | 13 | from distutils.version import StrictVersion
|
15 | 14 | from pandas import compat
|
@@ -387,23 +386,23 @@ def test_import_google_api_python_client(self):
|
387 | 386 |
|
388 | 387 | def test_should_return_bigquery_integers_as_python_ints(self):
|
389 | 388 | result = gbq._parse_entry(1, 'INTEGER')
|
390 |
| - tm.assert_equal(result, int(1)) |
| 389 | + assert result == int(1) |
391 | 390 |
|
392 | 391 | def test_should_return_bigquery_floats_as_python_floats(self):
|
393 | 392 | result = gbq._parse_entry(1, 'FLOAT')
|
394 |
| - tm.assert_equal(result, float(1)) |
| 393 | + assert result == float(1) |
395 | 394 |
|
396 | 395 | def test_should_return_bigquery_timestamps_as_numpy_datetime(self):
|
397 | 396 | result = gbq._parse_entry('0e9', 'TIMESTAMP')
|
398 |
| - tm.assert_equal(result, np_datetime64_compat('1970-01-01T00:00:00Z')) |
| 397 | + assert result == np_datetime64_compat('1970-01-01T00:00:00Z') |
399 | 398 |
|
400 | 399 | def test_should_return_bigquery_booleans_as_python_booleans(self):
|
401 | 400 | result = gbq._parse_entry('false', 'BOOLEAN')
|
402 |
| - tm.assert_equal(result, False) |
| 401 | + assert not result |
403 | 402 |
|
404 | 403 | def test_should_return_bigquery_strings_as_python_strings(self):
|
405 | 404 | result = gbq._parse_entry('STRING', 'STRING')
|
406 |
| - tm.assert_equal(result, 'STRING') |
| 405 | + assert result == 'STRING' |
407 | 406 |
|
408 | 407 | def test_to_gbq_should_fail_if_invalid_table_name_passed(self):
|
409 | 408 | with pytest.raises(gbq.NotFoundException):
|
@@ -709,7 +708,7 @@ def test_index_column(self):
|
709 | 708 | private_key=_get_private_key_path())
|
710 | 709 | correct_frame = DataFrame(
|
711 | 710 | {'string_1': ['a'], 'string_2': ['b']}).set_index("string_1")
|
712 |
| - tm.assert_equal(result_frame.index.name, correct_frame.index.name) |
| 711 | + assert result_frame.index.name == correct_frame.index.name |
713 | 712 |
|
714 | 713 | def test_column_order(self):
|
715 | 714 | query = "SELECT 'a' AS string_1, 'b' AS string_2, 'c' AS string_3"
|
|
0 commit comments