Skip to content

Commit 3db45a9

Browse files
committed
update tests for both lists and numpy array 1D and 2D
1 parent 662c0ff commit 3db45a9

File tree

1 file changed

+15
-2
lines changed

1 file changed

+15
-2
lines changed

pandas/tests/series/test_analytics.py

+15-2
Original file line numberDiff line numberDiff line change
@@ -849,17 +849,30 @@ def test_matmul(self):
849849
expected = np.dot(a.values, a.values)
850850
assert_almost_equal(result, expected)
851851

852-
# np.array @ Series (__rmatmul__)
852+
# GH 21530
853+
# vector (1D np.array) @ Series (__rmatmul__)
853854
result = operator.matmul(a.values, a)
854855
expected = np.dot(a.values, a.values)
855856
assert_almost_equal(result, expected)
856857

857858
# GH 21530
858-
# np.array (matrix) @ Series (__rmatmul__)
859+
# vector (1D list) @ Series (__rmatmul__)
860+
result = operator.matmul(a.tolist(), a)
861+
expected = np.dot(a.values, a.values)
862+
assert_almost_equal(result, expected)
863+
864+
# GH 21530
865+
# matrix (2D np.array) @ Series (__rmatmul__)
859866
result = operator.matmul(b.T.values, a)
860867
expected = np.dot(b.T.values, a.values)
861868
assert_almost_equal(result, expected)
862869

870+
# GH 21530
871+
# matrix (2D nested lists) @ Series (__rmatmul__)
872+
result = operator.matmul(b.T.tolist(), a)
873+
expected = np.dot(b.T.values, a.values)
874+
assert_almost_equal(result, expected)
875+
863876
# mixed dtype DataFrame @ Series
864877
a['p'] = int(a.p)
865878
result = operator.matmul(b.T, a)

0 commit comments

Comments
 (0)