Skip to content

Commit 5cd290d

Browse files
committed
Reduce query counts
1 parent 6c12227 commit 5cd290d

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

tests/integ/modin/frame/test_replace.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,9 @@ def snow_df():
2323
)
2424

2525

26-
@sql_count_checker(query_count=4)
27-
def test_replace_mauricio_dataframe():
28-
test_sdf = pd.DataFrame(
26+
@sql_count_checker(query_count=3)
27+
def test_replace_SNOW_2390752_dataframe():
28+
test_ndf = native_pd.DataFrame(
2929
{
3030
"col1": ["one", "two", "two", "three", "two", "four"],
3131
"col2": [
@@ -38,7 +38,7 @@ def test_replace_mauricio_dataframe():
3838
],
3939
}
4040
)
41-
test_ndf = test_sdf.to_pandas()
41+
test_sdf = pd.DataFrame(test_ndf)
4242
test_mdf = test_sdf.move_to("Pandas")
4343
assert test_mdf.get_backend() == "Pandas"
4444

@@ -50,11 +50,12 @@ def test_replace_mauricio_dataframe():
5050
assert modin_result.to_pandas().equals(snow_result.to_pandas())
5151

5252

53-
@sql_count_checker(query_count=4, join_count=2)
54-
def test_replace_mauricio_series():
55-
56-
input_dict = pd.Series({"a": "apple", "socks": "socks are darned", "b": "bee"})
57-
input_dict_native = input_dict.to_pandas()
53+
@sql_count_checker(query_count=3, join_count=2)
54+
def test_replace_SNOW_2390752_series():
55+
input_dict_native = native_pd.Series(
56+
{"a": "apple", "socks": "socks are darned", "b": "bee"}
57+
)
58+
input_dict = pd.Series(input_dict_native)
5859
input_dict_pandas_modin = input_dict.move_to("Pandas")
5960
assert input_dict_pandas_modin.get_backend() == "Pandas"
6061
expected = input_dict_native["socks"].replace("darned", "clean")

0 commit comments

Comments
 (0)