address comments

Author: sfc-gh-helmeleegy

tests/integ/modin/frame/test_pop.py

@@ -9,6 +9,7 @@
 import numpy as np
 import modin.pandas as pd
 import pandas as native_pd
+import pytest
 
 from tests.integ.modin.utils import assert_frame_equal, assert_series_equal
 from tests.integ.utils.sql_counter import sql_count_checker
@@ -32,3 +33,20 @@ def test_pop():
 
     assert_series_equal(snow_popped_ser, native_popped_ser)
     assert_frame_equal(snow_df, native_df)
+
+
+@sql_count_checker(query_count=0)
+def test_pop_not_found():
+    native_df = native_pd.DataFrame(
+        [
+            ("falcon", "bird", 389.0),
+            ("parrot", "bird", 24.0),
+            ("lion", "mammal", 80.5),
+            ("monkey", "mammal", np.nan),
+        ],
+        columns=("name", "class", "max_speed"),
+    )
+    snow_df = pd.DataFrame(native_df)
+
+    with pytest.raises(KeyError):
+        snow_df.pop("not_found")

tests/integ/modin/series/test_pop.py

@@ -8,6 +8,7 @@
 
 import modin.pandas as pd
 import pandas as native_pd
+import pytest
 
 from tests.integ.modin.utils import assert_series_equal
 from tests.integ.utils.sql_counter import sql_count_checker
@@ -24,3 +25,13 @@ def test_pop():
 
     assert snow_popped_val == native_popped_val
     assert_series_equal(snow_ser, native_ser)
+
+
+@sql_count_checker(query_count=2, join_count=1)
+def test_pop_not_found():
+    native_ser = native_pd.Series([1, 2, 3])
+    snow_ser = pd.Series(native_ser)
+    assert isinstance(snow_ser, pd.Series)
+
+    with pytest.raises(KeyError):
+        snow_ser.pop(-1)