47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
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 |