feat: better state_handler
This commit is contained in:
47
tests/state/test_state_handler.py
Normal file
47
tests/state/test_state_handler.py
Normal file
@@ -0,0 +1,47 @@
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
from diceplayer.config import PlayerConfig, DiceConfig, GaussianConfig
|
||||
from diceplayer.state.state_handler import StateHandler
|
||||
|
||||
|
||||
class TestStateHandler:
|
||||
@pytest.fixture
|
||||
def player_config(self) -> PlayerConfig:
|
||||
return PlayerConfig(
|
||||
opt=True,
|
||||
mem=12,
|
||||
maxcyc=100,
|
||||
nprocs=4,
|
||||
ncores=4,
|
||||
dice=DiceConfig(
|
||||
ljname="test",
|
||||
outname="test",
|
||||
dens=1.0,
|
||||
nmol=[1],
|
||||
nstep=[1, 1],
|
||||
),
|
||||
gaussian=GaussianConfig(
|
||||
level="test",
|
||||
qmprog="g16",
|
||||
keywords="test",
|
||||
),
|
||||
)
|
||||
|
||||
def test_state_handler_initialization(self):
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
tmpdir_path = Path(tmpdir)
|
||||
state_handler = StateHandler(tmpdir_path)
|
||||
|
||||
assert isinstance(state_handler, StateHandler)
|
||||
|
||||
def test_state_handler_get_state(self, player_config: PlayerConfig):
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
tmpdir_path = Path(tmpdir)
|
||||
state_handler = StateHandler(tmpdir_path)
|
||||
|
||||
state = state_handler.get_state(player_config)
|
||||
|
||||
assert state is None
|
||||
Reference in New Issue
Block a user