Skip to content

Commit eb8e42d

Browse files
committed
Fix test
1 parent fadfdf7 commit eb8e42d

File tree

1 file changed

+1
-25
lines changed

1 file changed

+1
-25
lines changed

tests/unit/test_distributed.py

Lines changed: 1 addition & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import pytest
66

77
import awswrangler as wr
8-
from awswrangler._distributed import EngineEnum, MemoryFormatEnum
8+
from awswrangler._distributed import EngineEnum
99
from awswrangler.s3._write_parquet import _to_parquet
1010

1111
logging.getLogger("awswrangler").setLevel(logging.DEBUG)
@@ -28,27 +28,3 @@ def test_engine(engine_enum: Enum) -> None:
2828
assert wr.engine._registry
2929
assert wr.engine.dispatch_func(_to_parquet).__name__.endswith("distributed")
3030
assert not wr.engine.dispatch_func(_to_parquet, "python").__name__.endswith("distributed")
31-
32-
wr.engine.register("python")
33-
assert wr.engine.get_installed() == engine_enum
34-
assert wr.engine.get() == EngineEnum.PYTHON
35-
assert not wr.engine._registry
36-
assert not wr.engine.dispatch_func(_to_parquet).__name__.endswith("distributed")
37-
38-
39-
@pytest.mark.parametrize(
40-
"memory_format_enum",
41-
[
42-
pytest.param(
43-
MemoryFormatEnum.MODIN,
44-
marks=pytest.mark.skip("modin not available") if not importlib.util.find_spec("modin") else [],
45-
),
46-
],
47-
)
48-
def test_memory_format(memory_format_enum: Enum) -> None:
49-
assert wr.memory_format.get_installed() == memory_format_enum
50-
assert wr.memory_format.get() == memory_format_enum
51-
52-
wr.memory_format.set("pandas")
53-
assert wr.memory_format.get_installed() == memory_format_enum
54-
assert wr.memory_format.get() == MemoryFormatEnum.PANDAS

0 commit comments

Comments
 (0)