Skip to content

Commit 475f069

Browse files
update engine sub package imports
1 parent 570cd7b commit 475f069

3 files changed

Lines changed: 9 additions & 4 deletions

File tree

chmengine/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
"""A silly chess engine that picks moves using heatmaps"""
22
from chmengine import engines, play, utils
3-
from chmengine.engines import CMHMEngine, CMHMEngine2, Quartney, CMHMEngine2PoolExecutor
3+
from chmengine.engines import CMHMEngine, CMHMEngine2, Quartney, CMHMEngine2PoolExecutor, CMHMEngine3
44
from chmengine.play import PlayCMHMEngine
55
from chmengine.utils import (Pick, better_checkmate_score, calculate_better_white_minus_black_score,
66
calculate_white_minus_black_score, checkmate_score, format_moves, format_picks,
@@ -20,6 +20,7 @@
2020
'PlayCMHMEngine',
2121
'Pick',
2222
'CMHMEngine2PoolExecutor',
23+
'CMHMEngine3',
2324
# functions
2425
'format_moves',
2526
'format_picks',

chmengine/engines/__init__.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,22 @@
11
"""Engines"""
2-
from chmengine.engines import cmhmey1, cmhmey2, quartney, cmhmey2_pool_executor
2+
from chmengine.engines import cmhmey1, cmhmey2, quartney, cmhmey2_pool_executor, cmhmey3
33
from chmengine.engines.cmhmey1 import CMHMEngine
44
from chmengine.engines.cmhmey2 import CMHMEngine2
55
from chmengine.engines.quartney import Quartney
66
from chmengine.engines.cmhmey2_pool_executor import CMHMEngine2PoolExecutor
7+
from chmengine.engines.cmhmey3 import CMHMEngine3
78

89
__all__ = [
910
# Mods
1011
'cmhmey1',
1112
'cmhmey2',
1213
'quartney',
1314
'cmhmey2_pool_executor',
15+
'cmhmey3',
1416
# Classes
1517
'CMHMEngine',
1618
'CMHMEngine2',
1719
'Quartney',
1820
'CMHMEngine2PoolExecutor',
21+
'CMHMEngine3',
1922
]

tests/test_chmengine.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ class TestCMHEngineImports(TestCase):
1111
_engine1 = ['CMHMEngine']
1212
_engine2 = ['CMHMEngine2']
1313
_engine3 = ['Quartney']
14+
_engine4 = ['CMHMEngine3']
1415
_poole = ['CMHMEngine2PoolExecutor']
1516
_engine_manager = ['PlayCMHMEngine']
1617
_pick = ['Pick']
@@ -34,8 +35,8 @@ class TestCMHEngineImports(TestCase):
3435
'get_static_value',
3536
'max_moves_map',
3637
]
37-
_engine_mods = ['cmhmey1', 'cmhmey2', 'quartney', 'cmhmey2_pool_executor']
38-
_engines = _engine1 + _engine2 + _engine3 + _poole
38+
_engine_mods = ['cmhmey1', 'cmhmey2', 'cmhmey3', 'quartney', 'cmhmey2_pool_executor']
39+
_engines = _engine1 + _engine2 + _engine3 + _poole + _engine4
3940
_all = _engines + _engine_manager + _pick + _functions + _modules
4041

4142
def setUp(self) -> None:

0 commit comments

Comments
 (0)