@@ -6039,46 +6039,47 @@ def test_combineSeries(self):
6039
6039
#added = self.mixed_int + (100*series).astype('int32')
6040
6040
#_check_mixed_int(added, dtype = dict(A = 'int32', B = 'float64', C = 'int32', D = 'int64'))
6041
6041
6042
- # TimeSeries
6043
- buf = StringIO()
6044
- tmp = sys.stderr
6045
- sys.stderr = buf
6046
6042
6047
- try:
6048
- ts = self.tsframe['A']
6049
- added = self.tsframe + ts
6050
-
6051
- for key, col in compat.iteritems(self.tsframe):
6052
- result = col + ts
6053
- assert_series_equal(added[key], result, check_names=False)
6054
- self.assertEqual(added[key].name, key)
6055
- if col.name == ts.name:
6056
- self.assertEqual(result.name, 'A')
6057
- else:
6058
- self.assertTrue(result.name is None)
6043
+ # TimeSeries
6044
+ ts = self.tsframe['A']
6045
+
6046
+ # 10890
6047
+ # we no longer allow auto timeseries broadcasting
6048
+ # and require explict broadcasting
6049
+ added = self.tsframe.add(ts, axis='index')
6050
+
6051
+ for key, col in compat.iteritems(self.tsframe):
6052
+ result = col + ts
6053
+ assert_series_equal(added[key], result, check_names=False)
6054
+ self.assertEqual(added[key].name, key)
6055
+ if col.name == ts.name:
6056
+ self.assertEqual(result.name, 'A')
6057
+ else:
6058
+ self.assertTrue(result.name is None)
6059
6059
6060
- smaller_frame = self.tsframe[:-5]
6061
- smaller_added = smaller_frame + ts
6060
+ smaller_frame = self.tsframe[:-5]
6061
+ smaller_added = smaller_frame.add(ts, axis='index')
6062
6062
6063
- self.assertTrue(smaller_added.index.equals(self.tsframe.index))
6063
+ self.assertTrue(smaller_added.index.equals(self.tsframe.index))
6064
6064
6065
- smaller_ts = ts[:-5]
6066
- smaller_added2 = self.tsframe + smaller_ts
6067
- assert_frame_equal(smaller_added, smaller_added2)
6065
+ smaller_ts = ts[:-5]
6066
+ smaller_added2 = self.tsframe.add( smaller_ts, axis='index')
6067
+ assert_frame_equal(smaller_added, smaller_added2)
6068
6068
6069
- # length 0
6070
- result = self.tsframe + ts[:0]
6069
+ # length 0, result is all-nan
6070
+ result = self.tsframe.add(ts[:0], axis='index')
6071
+ expected = DataFrame(np.nan,index=self.tsframe.index,columns=self.tsframe.columns)
6072
+ assert_frame_equal(result, expected)
6071
6073
6072
- # Frame is length 0
6073
- result = self.tsframe[:0] + ts
6074
- self.assertEqual(len(result), 0)
6074
+ # Frame is all-nan
6075
+ result = self.tsframe[:0].add(ts, axis='index')
6076
+ expected = DataFrame(np.nan,index=self.tsframe.index,columns=self.tsframe.columns)
6077
+ assert_frame_equal(result, expected)
6075
6078
6076
- # empty but with non-empty index
6077
- frame = self.tsframe[:1].reindex(columns=[])
6078
- result = frame * ts
6079
- self.assertEqual(len(result), len(ts))
6080
- finally:
6081
- sys.stderr = tmp
6079
+ # empty but with non-empty index
6080
+ frame = self.tsframe[:1].reindex(columns=[])
6081
+ result = frame.mul(ts,axis='index')
6082
+ self.assertEqual(len(result), len(ts))
6082
6083
6083
6084
def test_combineFunc(self):
6084
6085
result = self.frame * 2
0 commit comments