Coverage for tests/unit/test_defaults.py: 100%

18 statements  

« prev     ^ index     » next       coverage.py v7.4.4, created at 2024-04-24 22:41 -0600

1"""Test functions in es_testbed.defaults""" 

2# pylint: disable=missing-function-docstring,redefined-outer-name 

3import pytest 

4from es_testbed.defaults import ilmcold, ilmwarm, ilm_force_merge, ilm_phase 

5 

6@pytest.fixture 

7def forcemerge(): 

8 def _forcemerge(mns: int=1): 

9 return {'forcemerge': {'max_num_segments': mns}} 

10 return _forcemerge 

11 

12def test_default_ilm_fm(forcemerge): 

13 assert ilm_force_merge() == forcemerge() 

14 

15def test_default_ilm_fm_mns(forcemerge): 

16 mns = 2 

17 assert ilm_force_merge(max_num_segments=mns) == forcemerge(mns=mns) 

18 

19def test_default_ilm_warm(): 

20 tier = 'warm' 

21 assert ilm_phase(tier) == {tier: ilmwarm()} 

22 

23def test_default_ilm_cold(): 

24 tier = 'cold' 

25 assert ilm_phase(tier) == {tier: ilmcold()}