|
16 | 16 | timedelta_range,
|
17 | 17 | )
|
18 | 18 | import pandas.util.testing as tm
|
19 |
| -from pandas.util.testing import ( |
20 |
| - assert_almost_equal, |
21 |
| - assert_frame_equal, |
22 |
| - assert_series_equal, |
23 |
| -) |
24 | 19 |
|
25 | 20 |
|
26 | 21 | class SharedWithSparse:
|
@@ -344,7 +339,7 @@ def test_values(self, float_frame, float_string_frame):
|
344 | 339 | # single block corner case
|
345 | 340 | arr = float_frame[["A", "B"]].values
|
346 | 341 | expected = float_frame.reindex(columns=["A", "B"]).values
|
347 |
| - assert_almost_equal(arr, expected) |
| 342 | + tm.assert_almost_equal(arr, expected) |
348 | 343 |
|
349 | 344 | def test_to_numpy(self):
|
350 | 345 | df = pd.DataFrame({"A": [1, 2], "B": [3, 4.5]})
|
@@ -401,11 +396,11 @@ def test_axis_aliases(self, float_frame):
|
401 | 396 | # reg name
|
402 | 397 | expected = f.sum(axis=0)
|
403 | 398 | result = f.sum(axis="index")
|
404 |
| - assert_series_equal(result, expected) |
| 399 | + tm.assert_series_equal(result, expected) |
405 | 400 |
|
406 | 401 | expected = f.sum(axis=1)
|
407 | 402 | result = f.sum(axis="columns")
|
408 |
| - assert_series_equal(result, expected) |
| 403 | + tm.assert_series_equal(result, expected) |
409 | 404 |
|
410 | 405 | def test_class_axis(self):
|
411 | 406 | # GH 18147
|
@@ -474,8 +469,8 @@ class TestDataFrameMisc(SharedWithSparse):
|
474 | 469 |
|
475 | 470 | klass = DataFrame
|
476 | 471 | # SharedWithSparse tests use generic, klass-agnostic assertion
|
477 |
| - _assert_frame_equal = staticmethod(assert_frame_equal) |
478 |
| - _assert_series_equal = staticmethod(assert_series_equal) |
| 472 | + _assert_frame_equal = staticmethod(tm.assert_frame_equal) |
| 473 | + _assert_series_equal = staticmethod(tm.assert_series_equal) |
479 | 474 |
|
480 | 475 | def test_values(self, float_frame):
|
481 | 476 | float_frame.values[:, 0] = 5.0
|
|
0 commit comments