Skip to content

Commit 2b8072e

Browse files
committed
TST: fixes arithmetic tests for object output of period math
1 parent 11e150d commit 2b8072e

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

pandas/tests/frame/test_arithmetic.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ def test_ops_frame_period(self):
245245
exp = pd.DataFrame({'A': np.array([2, 1], dtype=object),
246246
'B': np.array([14, 13], dtype=object)})
247247
tm.assert_frame_equal(p - df, exp)
248-
tm.assert_frame_equal(df - p, -exp)
248+
tm.assert_frame_equal(df - p, -1 * exp)
249249

250250
df2 = pd.DataFrame({'A': [pd.Period('2015-05', freq='M'),
251251
pd.Period('2015-06', freq='M')],
@@ -257,4 +257,4 @@ def test_ops_frame_period(self):
257257
exp = pd.DataFrame({'A': np.array([4, 4], dtype=object),
258258
'B': np.array([16, 16], dtype=object)})
259259
tm.assert_frame_equal(df2 - df, exp)
260-
tm.assert_frame_equal(df - df2, -exp)
260+
tm.assert_frame_equal(df - df2, -1 * exp)

pandas/tests/series/test_arithmetic.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -315,15 +315,15 @@ def test_ops_series_period(self):
315315
# dtype will be object because of original dtype
316316
expected = pd.Series([9, 8], name='xxx', dtype=object)
317317
tm.assert_series_equal(per - ser, expected)
318-
tm.assert_series_equal(ser - per, -expected)
318+
tm.assert_series_equal(ser - per, -1 * expected)
319319

320320
s2 = pd.Series([pd.Period('2015-01-05', freq='D'),
321321
pd.Period('2015-01-04', freq='D')], name='xxx')
322322
assert s2.dtype == object
323323

324324
expected = pd.Series([4, 2], name='xxx', dtype=object)
325325
tm.assert_series_equal(s2 - ser, expected)
326-
tm.assert_series_equal(ser - s2, -expected)
326+
tm.assert_series_equal(ser - s2, -1 * expected)
327327

328328

329329
class TestTimestampSeriesArithmetic(object):

0 commit comments

Comments
 (0)