|
6 | 6 | import nose
|
7 | 7 | import numpy as np
|
8 | 8 | import sys
|
9 |
| - |
| 9 | +from pandas import Series |
10 | 10 | from pandas.util.testing import (
|
11 |
| - assert_almost_equal, assertRaisesRegexp, raise_with_traceback |
| 11 | + assert_almost_equal, assertRaisesRegexp, raise_with_traceback, assert_series_equal |
12 | 12 | )
|
13 | 13 |
|
14 | 14 | # let's get meta.
|
@@ -127,3 +127,29 @@ def test_raise_with_traceback(self):
|
127 | 127 | e = LookupError("error_text")
|
128 | 128 | _, _, traceback = sys.exc_info()
|
129 | 129 | raise_with_traceback(e, traceback)
|
| 130 | + |
| 131 | +class TestAssertSeriesEqual(unittest.TestCase): |
| 132 | + _multiprocess_can_split_ = True |
| 133 | + |
| 134 | + def _assert_equal(self, x, y, **kwargs): |
| 135 | + assert_series_equal(x,y,**kwargs) |
| 136 | + assert_series_equal(y,x,**kwargs) |
| 137 | + |
| 138 | + def _assert_not_equal(self, a, b, **kwargs): |
| 139 | + self.assertRaises(AssertionError, assert_series_equal, a, b, **kwargs) |
| 140 | + self.assertRaises(AssertionError, assert_series_equal, b, a, **kwargs) |
| 141 | + |
| 142 | + def test_equal(self): |
| 143 | + self._assert_equal(Series(range(3)),Series(range(3))) |
| 144 | + self._assert_equal(Series(list('abc')),Series(list('abc'))) |
| 145 | + |
| 146 | + def test_not_equal(self): |
| 147 | + self._assert_not_equal(Series(range(3)),Series(range(3))+1) |
| 148 | + self._assert_not_equal(Series(list('abc')),Series(list('xyz'))) |
| 149 | + self._assert_not_equal(Series(range(3)),Series(range(4))) |
| 150 | + self._assert_not_equal(Series(range(3)),Series(range(3),dtype='float64')) |
| 151 | + self._assert_not_equal(Series(range(3)),Series(range(3),index=[1,2,4])) |
| 152 | + |
| 153 | + # ATM meta data is not checked in assert_series_equal |
| 154 | + # self._assert_not_equal(Series(range(3)),Series(range(3),name='foo'),check_names=True) |
| 155 | + |
0 commit comments