|
15 | 15 | from numpy.random import randn
|
16 | 16 | import numpy as np
|
17 | 17 |
|
18 |
| -from pandas import DataFrame, Series, date_range, timedelta_range, Categorical |
| 18 | +from pandas import (DataFrame, Series, date_range, timedelta_range, |
| 19 | + Categorical, SparseDataFrame) |
19 | 20 | import pandas as pd
|
20 | 21 |
|
21 | 22 | from pandas.util.testing import (assert_almost_equal,
|
@@ -214,6 +215,18 @@ def test_iterrows(self):
|
214 | 215 | exp = self.mixed_frame.loc[k]
|
215 | 216 | self._assert_series_equal(v, exp)
|
216 | 217 |
|
| 218 | + def test_iterrows_iso8601(self): |
| 219 | + # GH19671 |
| 220 | + if self.klass == SparseDataFrame: |
| 221 | + pytest.xfail(reason='SparseBlock datetime type not implemented.') |
| 222 | + |
| 223 | + s = self.klass( |
| 224 | + {'non_iso8601': ['M1701', 'M1802', 'M1903', 'M2004'], |
| 225 | + 'iso8601': date_range('2000-01-01', periods=4, freq='M')}) |
| 226 | + for k, v in s.iterrows(): |
| 227 | + exp = s.loc[k] |
| 228 | + self._assert_series_equal(v, exp) |
| 229 | + |
217 | 230 | def test_itertuples(self):
|
218 | 231 | for i, tup in enumerate(self.frame.itertuples()):
|
219 | 232 | s = self.klass._constructor_sliced(tup[1:])
|
|
0 commit comments