feat: better state_handler

This commit is contained in:
2026-03-04 14:02:19 -03:00
parent 11ff4c0c21
commit 06ae9b41f0
17 changed files with 158 additions and 42 deletions

View 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