Coverage for mlos_bench/mlos_bench/tests/storage/conftest.py: 100%

1# 

2# Copyright (c) Microsoft Corporation. 

3# Licensed under the MIT License. 

4# 

5"""Export test fixtures for mlos_bench storage.""" 

6 

7import mlos_bench.tests.storage.sql.fixtures as sql_storage_fixtures 

8 

9# NOTE: For future storage implementation additions, we can refactor this to use 

10# lazy_fixture and parameterize the tests across fixtures but keep the test code the 

11# same. 

12 

13# Expose some of those as local names so they can be picked up as fixtures by pytest. 

14storage = sql_storage_fixtures.storage 

15exp_storage = sql_storage_fixtures.exp_storage 

16exp_no_tunables_storage = sql_storage_fixtures.exp_no_tunables_storage 

17mixed_numerics_exp_storage = sql_storage_fixtures.mixed_numerics_exp_storage 

18exp_storage_with_trials = sql_storage_fixtures.exp_storage_with_trials 

19exp_no_tunables_storage_with_trials = sql_storage_fixtures.exp_no_tunables_storage_with_trials 

20mixed_numerics_exp_storage_with_trials = ( 

21 sql_storage_fixtures.mixed_numerics_exp_storage_with_trials 

22) 

23exp_data = sql_storage_fixtures.exp_data 

24exp_no_tunables_data = sql_storage_fixtures.exp_no_tunables_data 

25mixed_numerics_exp_data = sql_storage_fixtures.mixed_numerics_exp_data