def test_init(options, shellmound_ModflowSfr2):
mf6sfr = sfrmaker.Mf6SFR(shellmound_ModflowSfr2, options=options)
options_block = mf6sfr.options_block.strip('\n').split('\n')
expected_keys = ['BEGIN']
if options is not None:
expected_keys = expected_keys[:1] + [
' '.join(item.split()[:-1]) if len(item.split()) > 1 else item
for item in options
]
if 'unit_conversion' not in expected_keys:
expected_keys.append('unit_conversion')
if 'auxiliary' not in expected_keys:
expected_keys.append('auxiliary')
expected_keys.append('END')
keys = [
' '.join(item.split()[:-1]) if len(item.split()) > 1 else item.strip()
for item in options_block
]
# verify that there aren't any duplicate entries
assert len(set(keys)) == len(keys)
assert keys == expected_keys
def mf6sfr_instance_SFRdata(shellmound_sfrdata):
return sfrmaker.Mf6SFR(SFRData=shellmound_sfrdata)
def mf6sfr_instance_ModflowSfr2(shellmound_ModflowSfr2):
return sfrmaker.Mf6SFR(shellmound_ModflowSfr2)