5
5
import pytest
6
6
7
7
import awswrangler as wr
8
- from awswrangler ._distributed import EngineEnum , MemoryFormatEnum
8
+ from awswrangler ._distributed import EngineEnum
9
9
from awswrangler .s3 ._write_parquet import _to_parquet
10
10
11
11
logging .getLogger ("awswrangler" ).setLevel (logging .DEBUG )
@@ -28,27 +28,3 @@ def test_engine(engine_enum: Enum) -> None:
28
28
assert wr .engine ._registry
29
29
assert wr .engine .dispatch_func (_to_parquet ).__name__ .endswith ("distributed" )
30
30
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