@@ -6039,46 +6039,47 @@ def test_combineSeries(self):
60396039 #added = self.mixed_int + (100*series).astype('int32')
60406040 #_check_mixed_int(added, dtype = dict(A = 'int32', B = 'float64', C = 'int32', D = 'int64'))
60416041
6042- # TimeSeries
6043- buf = StringIO()
6044- tmp = sys.stderr
6045- sys.stderr = buf
60466042
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)
60596059
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')
60626062
6063- self.assertTrue(smaller_added.index.equals(self.tsframe.index))
6063+ self.assertTrue(smaller_added.index.equals(self.tsframe.index))
60646064
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)
60686068
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)
60716073
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)
60756078
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))
60826083
60836084 def test_combineFunc(self):
60846085 result = self.frame * 2
0 commit comments