def test_build_parameter_record():
name = "ssr2gw_rate 0"
values = np.random.randn(128)
dimensions = [["nssr", 128]]
datatype = 2
pr = ParameterRecord(name,
values=values,
dimensions=dimensions,
datatype=datatype)
assert pr.ndim == 1
assert pr.name == name
assert np.allclose(values, pr.values)
assert pr.datatype == datatype
def test_build_prms_parameter_object():
file_name = "test.param"
name = "ssr2gw_rate 0"
values = np.random.randn(128)
dimensions = [["nssr", 128]]
datatype = 2
pr = ParameterRecord(name, values=values, dimensions=dimensions,
datatype=datatype, file_name=file_name)
pp = PrmsParameters([pr], header="this is a test")
assert pp.parameter_files[0] == file_name
assert pp.record_names[0] == name
assert isinstance(pp.parameters_list[0], ParameterRecord)
assert np.allclose(pp.parameters_list[0].values, pr.values)
def test_build_prms_model():
# buld control file object
cf_name = "test.control"
model_dir = os.path.join(ws, ".")
name = "model_mode"
datatype = 4
cr_values = ["GSFLOW"]
cr = ControlRecord(name=name, values=cr_values, datatype=datatype)
control = ControlFile([cr],
name=cf_name,
model_dir=model_dir,
header="this is a test")
# build parameter object
file_name = "test.param"
name = "ssr2gw_rate 0"
values = np.random.randn(128)
dimensions = [["nssr", 128]]
datatype = 2
pr = ParameterRecord(name,
values=values,
dimensions=dimensions,
datatype=datatype,
file_name=file_name)
parameters = PrmsParameters([pr], header="this is a test")
# build Data object
data_df = pd.DataFrame()
name = "test.data"
model_dir = os.path.join(ws, ".")
header = "this is a test"
data = PrmsData(data_df, name=name, model_dir=model_dir, header=header)
prms = PrmsModel(control, parameters=parameters, data=data)
assert isinstance(prms.control, ControlFile)
assert isinstance(prms.parameters, PrmsParameters)
assert isinstance(prms.data, PrmsData)
assert prms.control_file == os.path.join(model_dir, cf_name)
assert prms.parameters == parameters
assert prms.data == data
assert prms.control == control
assert np.allclose(prms.parameters.parameters_list[0].values, values)
assert prms.control.records_list[0].values[0] == cr_values[0]
def test_empty_parameter_record():
pr = ParameterRecord(name="test", values=[1])
assert isinstance(pr, ParameterRecord)
assert isinstance(pr, RecordBase)