def test_single_ri_folder(self, standard, il):
with patch('oasislmf.model_execution.bin.INPUT_FILES',
ECHO_CONVERSION_INPUT_FILES), TemporaryDirectory(
) as csv_dir, TemporaryDirectory() as bin_dir:
files = standard + il
for target in files:
with io.open(os.path.join(csv_dir, target + '.csv'),
'w',
encoding='utf-8') as f:
f.write(target)
os.mkdir(os.path.join(csv_dir, "RI_1"))
for target in files:
with io.open(os.path.join(csv_dir, "RI_1", target + '.csv'),
'w',
encoding='utf-8') as f:
f.write(target)
csv_to_bin(csv_dir, bin_dir, il=True, ri=True)
self.assertEqual(len(files),
len(glob.glob(os.path.join(bin_dir, '*.bin'))))
for filename in (f + '.bin' for f in files):
self.assertTrue(os.path.exists(os.path.join(bin_dir,
filename)))
self.assertEqual(
len(files),
len(
glob.glob(
os.path.join(bin_dir, 'RI_1{}*.bin'.format(os.sep)))))
for filename in (f + '.bin' for f in files):
self.assertTrue(
os.path.exists(os.path.join(bin_dir, 'RI_1', filename)))
print("ok")
def test_subprocess_raises___oasis_exception_is_raised(self):
with TemporaryDirectory() as csv_dir, TemporaryDirectory() as bin_dir:
Path(os.path.join(csv_dir, 'events.csv')).touch()
with patch('oasislmf.model_execution.bin.subprocess.check_call', Mock(side_effect=subprocess.CalledProcessError(1, ''))):
with self.assertRaises(OasisException):
csv_to_bin(csv_dir, bin_dir, il=True)
def test_directory_only_contains_excluded_files___tar_is_empty(self):
with TemporaryDirectory() as csv_dir, TemporaryDirectory() as bin_dir:
with io_open(os.path.join(csv_dir, 'another_file'), 'w', encoding='utf-8') as f:
f.write('file data')
csv_to_bin(csv_dir, bin_dir)
self.assertEqual(0, len(glob.glob(os.path.join(csv_dir, '*.bin'))))
def test_contains_il_and_standard_files_but_il_is_true___all_files_are_included(self, standard, il):
with patch('oasislmf.model_execution.bin.INPUT_FILES', ECHO_CONVERSION_INPUT_FILES), TemporaryDirectory() as csv_dir, TemporaryDirectory() as bin_dir:
for target in chain(standard, il):
with io_open(os.path.join(csv_dir, target + '.csv'), 'w', encoding='utf-8') as f:
f.write(target)
csv_to_bin(csv_dir, bin_dir, il=True)
self.assertEqual(len(standard) + len(il), len(glob.glob(os.path.join(bin_dir, '*.bin'))))
for filename in (f + '.bin' for f in chain(standard, il)):
self.assertTrue(os.path.exists(os.path.join(bin_dir, filename)))
def _run_test(self, account_df, location_df, ri_info_df, ri_scope_df,
loss_factor, do_reinsurance):
"""
Run the direct and reinsurance layers through the Oasis FM.
Returns an array of net loss data frames, the first for the direct layers
and then one per inuring layer.
"""
t_start = time.time()
net_losses = OrderedDict()
initial_dir = os.getcwd()
try:
with TemporaryDirectory() as run_dir:
os.chdir(run_dir)
direct_layer = DirectLayer(account_df, location_df)
direct_layer.generate_oasis_structures()
direct_layer.write_oasis_files()
losses_df = direct_layer.apply_fm(
loss_percentage_of_tiv=loss_factor, net=False)
net_losses['Direct'] = losses_df
oed_validator = oed.OedValidator()
if do_reinsurance:
(is_valid, error_msgs) = oed_validator.validate(
ri_info_df, ri_scope_df)
if not is_valid:
print(error_msgs)
exit(1)
ri_layers = reinsurance_layer.generate_files_for_reinsurance(
items=direct_layer.items,
coverages=direct_layer.coverages,
fm_xrefs=direct_layer.fm_xrefs,
xref_descriptions=direct_layer.xref_descriptions,
gulsummaryxref=pd.DataFrame(),
fmsummaryxref=pd.DataFrame(),
ri_info_df=ri_info_df,
ri_scope_df=ri_scope_df,
direct_oasis_files_dir='',
)
for idx in ri_layers:
'''
{'inuring_priority': 1, 'risk_level': 'LOC', 'directory': 'run/RI_1'}
{'inuring_priority': 1, 'risk_level': 'ACC', 'directory': 'run/RI_2'}
{'inuring_priority': 2, 'risk_level': 'LOC', 'directory': 'run/RI_3'}
{'inuring_priority': 3, 'risk_level': 'LOC', 'directory': 'run/RI_4'}
'''
if idx < 2:
input_name = "ils"
else:
input_name = ri_layers[idx - 1]['directory']
bin.csv_to_bin(ri_layers[idx]['directory'],
ri_layers[idx]['directory'],
il=True)
reinsurance_layer_losses_df = self._run_fm(
input_name, ri_layers[idx]['directory'],
direct_layer.xref_descriptions)
output_name = "Inuring_priority:{} - Risk_level:{}".format(
ri_layers[idx]['inuring_priority'],
ri_layers[idx]['risk_level'])
net_losses[output_name] = reinsurance_layer_losses_df
return net_losses
finally:
os.chdir(initial_dir)
t_end = time.time()
print("Exec time: {}".format(t_end - t_start))
