def test_store_params(casedir): pp = PostProcessor(dict(casedir=casedir)) params = ParamDict(Field=Field.default_params(), PostProcessor=PostProcessor.default_params()) pp.store_params(params) # Read back params params2 = None with open(os.path.join(pp.get_casedir(), "params.pickle"), 'r') as f: params2 = pickle.load(f) assert params2 == params str_params2 = open(os.path.join(pp.get_casedir(), "params.txt"), 'r').read() assert str_params2 == str(params)
def default_params(cls): params = PostProcessor.default_params() params.update(pp=2) return params