HideyoshiNakazone/DicePlayer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: fixes tests

Browse Source
This commit is contained in:
Vitor Hideyoshi
2026-02-26 06:23:30 -03:00
parent f56d6b6b36
commit e5c6282c86
18 changed files with 284 additions and 127 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
diceplayer/player.py
View File

@@ -1,7 +1,9 @@
from pydantic import BaseModel
from diceplayer import VERSION, logger
from diceplayer.shared.config.dice_config import DiceConfig
from diceplayer.shared.config.gaussian_config import GaussianDTO
from diceplayer.shared.config.player_config import PlayerConfig
from diceplayer.config.dice_config import DiceConfig
from diceplayer.config.gaussian_config import GaussianConfig
from diceplayer.config.player_config import PlayerConfig
from diceplayer.shared.environment.atom import Atom
from diceplayer.shared.environment.molecule import Molecule
from diceplayer.shared.environment.system import System
@@ -16,7 +18,6 @@ import yaml
import os
import pickle
import sys
from dataclasses import fields
from pathlib import Path
from typing import Tuple, Type
@@ -108,14 +109,16 @@ class Player:
geoms_file_path.touch()
def print_keywords(self) -> None:
def log_keywords(config: Dataclass, dto: Type[Dataclass]):
for key in sorted(list(map(lambda f: f.name, fields(dto)))):
if getattr(config, key) is not None:
if isinstance(getattr(config, key), list):
string = " ".join(str(x) for x in getattr(config, key))
logger.info(f"{key} = [ {string} ]")
else:
logger.info(f"{key} = {getattr(config, key)}")
def log_keywords(config: BaseModel):
for key, value in sorted(config.model_dump().items()):
if value is None:
continue
if isinstance(value, list):
string = " ".join(str(x) for x in value)
logger.info(f"{key} = [ {string} ]")
else:
logger.info(f"{key} = {value}")
logger.info(
f"##########################################################################################\n"
@@ -138,7 +141,7 @@ class Player:
"------------------------------------------------------------------------------------------\n"
)
log_keywords(self.config.dice, DiceConfig)
log_keywords(self.config.dice)
logger.info(
"------------------------------------------------------------------------------------------\n"
@@ -146,7 +149,7 @@ class Player:
"------------------------------------------------------------------------------------------\n"
)
log_keywords(self.config.gaussian, GaussianDTO)
log_keywords(self.config.gaussian)
logger.info("\n")