Skip to content

Commit 7bf740e

Browse files
author
Laurent Mutricy
committed
adding specific test
1 parent ee5e6c4 commit 7bf740e

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

pandas/tests/frame/test_query_eval.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,13 @@ def test_extension_array_eval(self, engine, parser, request):
214214
expected = Series(pd.array([0.25, 0.40, 0.50]))
215215
tm.assert_series_equal(result, expected)
216216

217+
def test_complex_eval(self, engine, parser):
218+
# GH#21374
219+
df = DataFrame({"a": [1 + 2j], "b": [1 + 1j]})
220+
result = df.eval("a/b", engine=engine, parser=parser)
221+
expected = Series([1.5 + 0.5j])
222+
tm.assert_series_equal(result, expected)
223+
217224

218225
class TestDataFrameQueryWithMultiIndex:
219226
def test_query_with_named_multiindex(self, parser, engine):

0 commit comments

Comments
 (0)