@@ -11787,61 +11787,65 @@ def test_update_from_non_df(self):
11787
11787
assert_frame_equal(df, expected)
11788
11788
11789
11789
def test_combineAdd(self):
11790
- # trivial
11791
- comb = self.frame.combineAdd(self.frame)
11792
- assert_frame_equal(comb, self.frame * 2)
11793
11790
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)
11830
11832
11831
11833
# TODO: test integer fill corner?
11832
11834
11833
11835
def test_combineMult(self):
11834
- # trivial
11835
- comb = self.frame.combineMult(self.frame)
11836
11836
11837
- assert_frame_equal(comb, self.frame ** 2)
11837
+ with tm.assert_produces_warning(FutureWarning):
11838
+ # trivial
11839
+ comb = self.frame.combineMult(self.frame)
11838
11840
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)
11842
11842
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)
11845
11849
11846
11850
def test_combine_generic(self):
11847
11851
df1 = self.frame
0 commit comments