Skip to content

Commit 4a3b605

Browse files
committed
fixed test case for cov
1 parent f0767f0 commit 4a3b605

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

Diff for: pandas/tests/frame/methods/test_cov_corr.py

+7-3
Original file line numberDiff line numberDiff line change
@@ -500,6 +500,10 @@ def test_corr_within_bounds(self):
500500

501501
def test_cov_with_missing_values(self):
502502
df = DataFrame({"A": [1, 2, None, 4], "B": [2, 4, None, 9]})
503-
expected = DataFrame({"A": [1.0, 1.0], "B": [1.0, 1.0]})
504-
result = df.dropna().cov()
505-
tm.assert_frame_equal(result, expected)
503+
expected = DataFrame(
504+
{"A": [2.333333, 5.500000], "B": [5.5, 13.0]}, index=["A", "B"]
505+
)
506+
result1 = df.cov()
507+
result2 = df.dropna().cov()
508+
tm.assert_frame_equal(result1, expected)
509+
tm.assert_frame_equal(result2, expected)

0 commit comments

Comments
 (0)