File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed
tests/core/engine_adapter/integration Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change 2222
2323from sqlmesh import Config , Context
2424from sqlmesh .cli .project_init import init_example_project
25+ from sqlmesh .core .config .common import VirtualEnvironmentMode
2526from sqlmesh .core .config .connection import ConnectionConfig
2627import sqlmesh .core .dialect as d
2728from sqlmesh .core .environment import EnvironmentSuffixTarget
@@ -1938,7 +1939,12 @@ def test_transaction(ctx: TestContext):
19381939 ctx .compare_with_current (table , input_data )
19391940
19401941
1941- def test_sushi (ctx : TestContext , tmp_path : pathlib .Path ):
1942+ @pytest .mark .parametrize (
1943+ "virtual_environment_mode" , [VirtualEnvironmentMode .FULL , VirtualEnvironmentMode .DEV_ONLY ]
1944+ )
1945+ def test_sushi (
1946+ ctx : TestContext , tmp_path : pathlib .Path , virtual_environment_mode : VirtualEnvironmentMode
1947+ ):
19421948 if ctx .mark == "athena_hive" :
19431949 pytest .skip (
19441950 "Sushi end-to-end tests only need to run once for Athena because sushi needs a hybrid of both Hive and Iceberg"
@@ -1984,6 +1990,7 @@ def _mutate_config(gateway: str, config: Config) -> None:
19841990 ).sql (dialect = config .model_defaults .dialect )
19851991 for e in before_all
19861992 ]
1993+ config .virtual_environment_mode = virtual_environment_mode
19871994
19881995 context = ctx .create_context (_mutate_config , path = tmp_path , ephemeral_state_connection = False )
19891996
You can’t perform that action at this time.
0 commit comments