71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
import pytest
|
|
import numpy as np
|
|
|
|
from damask import Config
|
|
from damask import Rotation
|
|
from damask import Orientation
|
|
|
|
class TestConfig:
|
|
|
|
@pytest.mark.parametrize('flow_style',[None,True,False])
|
|
def test_load_save_str(self,tmp_path,flow_style):
|
|
config = Config()
|
|
config['A'] = 1
|
|
config['B'] = [2,3]
|
|
config.save(tmp_path/'config.yaml',default_flow_style=flow_style)
|
|
assert Config.load(tmp_path/'config.yaml') == config
|
|
|
|
def test_load_save_file(self,tmp_path):
|
|
config = Config()
|
|
config['A'] = 1
|
|
config['B'] = [2,3]
|
|
with open(tmp_path/'config.yaml','w') as f:
|
|
config.save(f)
|
|
with open(tmp_path/'config.yaml') as f:
|
|
assert Config.load(f) == config
|
|
|
|
def test_add_remove(self):
|
|
dummy = {'hello':'world','foo':'bar'}
|
|
config = Config()
|
|
config |= dummy
|
|
assert config == Config() | dummy
|
|
config = config.delete(dummy)
|
|
assert config == Config()
|
|
assert (config | dummy ).delete( 'hello' ) == config | {'foo':'bar'}
|
|
assert (config | dummy ).delete([ 'hello', 'foo' ]) == config
|
|
assert (config | Config(dummy)).delete({ 'hello':1,'foo':2 }) == config
|
|
assert (config | Config(dummy)).delete(Config({'hello':1 })) == config | {'foo':'bar'}
|
|
|
|
|
|
def test_repr(self,tmp_path):
|
|
config = Config()
|
|
config['A'] = 1
|
|
config['B'] = [2,3]
|
|
with open(tmp_path/'config.yaml','w') as f:
|
|
f.write(config.__repr__())
|
|
assert Config.load(tmp_path/'config.yaml') == config
|
|
|
|
def test_numpy(self,tmp_path):
|
|
assert Config({'A':np.ones(3,'i'), 'B':np.ones(1)[0]}).__repr__() == \
|
|
Config({'A':[1,1,1], 'B':1.0}).__repr__()
|
|
|
|
def test_abstract_is_valid(self):
|
|
with pytest.raises(NotImplementedError):
|
|
Config().is_valid
|
|
|
|
def test_abstract_is_complete(self):
|
|
with pytest.raises(NotImplementedError):
|
|
Config().is_complete
|
|
|
|
@pytest.mark.parametrize('data',[Rotation.from_random(),Orientation.from_random(lattice='cI')])
|
|
def test_rotation_orientation(self,data):
|
|
assert str(Config(a=data)) == str(Config(a=data.as_quaternion()))
|
|
|
|
def test_initialize(self):
|
|
yml = """
|
|
a:
|
|
- 1
|
|
- 2
|
|
"""
|
|
assert Config(yml) == Config('{"a":[1,2]}') == Config(a=[1,2]) == Config(dict(a=[1,2]))
|