@@ -11787,61 +11787,65 @@ def test_update_from_non_df(self):
1178711787 assert_frame_equal(df, expected)
1178811788
1178911789 def test_combineAdd(self):
11790- # trivial
11791- comb = self.frame.combineAdd(self.frame)
11792- assert_frame_equal(comb, self.frame * 2)
1179311790
11794- # more rigorous
11795- a = DataFrame([[1., nan, nan, 2., nan]],
11796- columns=np.arange(5))
11797- b = DataFrame([[2., 3., nan, 2., 6., nan]],
11798- columns=np.arange(6))
11799- expected = DataFrame([[3., 3., nan, 4., 6., nan]],
11800- columns=np.arange(6))
11801-
11802- result = a.combineAdd(b)
11803- assert_frame_equal(result, expected)
11804- result2 = a.T.combineAdd(b.T)
11805- assert_frame_equal(result2, expected.T)
11806-
11807- expected2 = a.combine(b, operator.add, fill_value=0.)
11808- assert_frame_equal(expected, expected2)
11809-
11810- # corner cases
11811- comb = self.frame.combineAdd(self.empty)
11812- assert_frame_equal(comb, self.frame)
11813-
11814- comb = self.empty.combineAdd(self.frame)
11815- assert_frame_equal(comb, self.frame)
11816-
11817- # integer corner case
11818- df1 = DataFrame({'x': [5]})
11819- df2 = DataFrame({'x': [1]})
11820- df3 = DataFrame({'x': [6]})
11821- comb = df1.combineAdd(df2)
11822- assert_frame_equal(comb, df3)
11823-
11824- # mixed type GH2191
11825- df1 = DataFrame({'A': [1, 2], 'B': [3, 4]})
11826- df2 = DataFrame({'A': [1, 2], 'C': [5, 6]})
11827- rs = df1.combineAdd(df2)
11828- xp = DataFrame({'A': [2, 4], 'B': [3, 4.], 'C': [5, 6.]})
11829- assert_frame_equal(xp, rs)
11791+ with tm.assert_produces_warning(FutureWarning):
11792+ # trivial
11793+ comb = self.frame.combineAdd(self.frame)
11794+ assert_frame_equal(comb, self.frame * 2)
11795+
11796+ # more rigorous
11797+ a = DataFrame([[1., nan, nan, 2., nan]],
11798+ columns=np.arange(5))
11799+ b = DataFrame([[2., 3., nan, 2., 6., nan]],
11800+ columns=np.arange(6))
11801+ expected = DataFrame([[3., 3., nan, 4., 6., nan]],
11802+ columns=np.arange(6))
11803+
11804+ result = a.combineAdd(b)
11805+ assert_frame_equal(result, expected)
11806+ result2 = a.T.combineAdd(b.T)
11807+ assert_frame_equal(result2, expected.T)
11808+
11809+ expected2 = a.combine(b, operator.add, fill_value=0.)
11810+ assert_frame_equal(expected, expected2)
11811+
11812+ # corner cases
11813+ comb = self.frame.combineAdd(self.empty)
11814+ assert_frame_equal(comb, self.frame)
11815+
11816+ comb = self.empty.combineAdd(self.frame)
11817+ assert_frame_equal(comb, self.frame)
11818+
11819+ # integer corner case
11820+ df1 = DataFrame({'x': [5]})
11821+ df2 = DataFrame({'x': [1]})
11822+ df3 = DataFrame({'x': [6]})
11823+ comb = df1.combineAdd(df2)
11824+ assert_frame_equal(comb, df3)
11825+
11826+ # mixed type GH2191
11827+ df1 = DataFrame({'A': [1, 2], 'B': [3, 4]})
11828+ df2 = DataFrame({'A': [1, 2], 'C': [5, 6]})
11829+ rs = df1.combineAdd(df2)
11830+ xp = DataFrame({'A': [2, 4], 'B': [3, 4.], 'C': [5, 6.]})
11831+ assert_frame_equal(xp, rs)
1183011832
1183111833 # TODO: test integer fill corner?
1183211834
1183311835 def test_combineMult(self):
11834- # trivial
11835- comb = self.frame.combineMult(self.frame)
1183611836
11837- assert_frame_equal(comb, self.frame ** 2)
11837+ with tm.assert_produces_warning(FutureWarning):
11838+ # trivial
11839+ comb = self.frame.combineMult(self.frame)
1183811840
11839- # corner cases
11840- comb = self.frame.combineMult(self.empty)
11841- assert_frame_equal(comb, self.frame)
11841+ assert_frame_equal(comb, self.frame ** 2)
1184211842
11843- comb = self.empty.combineMult(self.frame)
11844- assert_frame_equal(comb, self.frame)
11843+ # corner cases
11844+ comb = self.frame.combineMult(self.empty)
11845+ assert_frame_equal(comb, self.frame)
11846+
11847+ comb = self.empty.combineMult(self.frame)
11848+ assert_frame_equal(comb, self.frame)
1184511849
1184611850 def test_combine_generic(self):
1184711851 df1 = self.frame
0 commit comments