Skip to content

Commit afe2d47

Browse files
committed
Add test
1 parent 6794d8d commit afe2d47

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

pandas/tests/test_frame.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15225,6 +15225,16 @@ def test_query_builtin(self):
1522515225
result = df.query('sin > 5', engine=engine, parser=parser)
1522615226
tm.assert_frame_equal(expected, result)
1522715227

15228+
def test_assignment_not_allowed(self):
15229+
df = DataFrame({'a': [1, 2, 3], 'b': ['a', 'b', 'c']})
15230+
a_before = df['a'].copy()
15231+
self.assertRaisesRegexp(
15232+
ValueError, 'assignment statement', df.query, 'a=1',
15233+
engine=self.engine, parser=self.parser
15234+
)
15235+
a_after = df['a'].copy()
15236+
assert_series_equal(a_before, a_after)
15237+
1522815238

1522915239
class TestDataFrameQueryPythonPython(TestDataFrameQueryNumExprPython):
1523015240

0 commit comments

Comments
 (0)