Skip to content

Commit 8faa85f

Browse files
Use 'test' in new backend and storage format names.
Signed-off-by: sfc-gh-mvashishtha <mahesh.vashishtha@snowflake.com>
1 parent 7738580 commit 8faa85f

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

modin/tests/config/test_envvars.py

+10-10
Original file line numberDiff line numberDiff line change
@@ -86,20 +86,20 @@ class CustomVar(cfg.EnvironmentVariable, type=request.param):
8686

8787

8888
@pytest.fixture(scope="session")
89-
def add_pandas_duplicate_4_on_ray_execution():
89+
def add_pandas_duplicate_on_ray_execution():
9090
"""
91-
Add an execution mode with the storage format Pandasduplicate4 and engine Ray.
91+
Add an execution mode with the storage format Test_Pandasduplicate and engine Ray.
9292
9393
This mode's execution is equivalent to PandasOnRay execution.
9494
"""
95-
cfg.StorageFormat.add_option("Pandasduplicate4")
95+
cfg.StorageFormat.add_option("Test_Pandasduplicate")
9696
from modin.core.execution.dispatching.factories import factories
9797

98-
factories.Pandasduplicate4OnRayFactory = factories.PandasOnRayFactory
98+
factories.Test_PandasduplicateOnRayFactory = factories.PandasOnRayFactory
9999
cfg.Backend.register_backend(
100-
"NewBackend4",
100+
"Test_Backend_1",
101101
cfg.Execution(
102-
storage_format="Pandasduplicate4",
102+
storage_format="Test_Pandasduplicate",
103103
engine="Ray",
104104
),
105105
)
@@ -652,22 +652,22 @@ def test_only_storage_format_comes_from_environment(
652652
self,
653653
clear_backend_execution_and_storage_format,
654654
order_to_get_in,
655-
add_pandas_duplicate_4_on_ray_execution,
655+
add_pandas_duplicate_on_ray_execution,
656656
):
657657
# To test switching StorageFormat alone, we have to add a new backend
658658
# that works with the default "Pandas" execution.
659659
with mock.patch.dict(
660660
os.environ,
661661
{
662-
cfg.StorageFormat.varname: "Pandasduplicate4",
662+
cfg.StorageFormat.varname: "Test_Pandasduplicate",
663663
},
664664
):
665665
cfg.Engine.put("Ray")
666666
for variable in order_to_get_in:
667667
expected_value = {
668-
cfg.Backend: "Newbackend4",
668+
cfg.Backend: "Test_Backend_1",
669669
cfg.Engine: "Ray",
670-
cfg.StorageFormat: "Pandasduplicate4",
670+
cfg.StorageFormat: "Test_Pandasduplicate",
671671
}[variable]
672672
assert (
673673
variable.get() == expected_value

0 commit comments

Comments
 (0)