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):
create_binary_files(csv_dir, bin_dir, do_il=True)
def test_multipl_ri_folders(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)
os.mkdir(os.path.join(csv_dir, "RI_2"))
for target in files:
with io.open(os.path.join(csv_dir, "RI_2", target + '.csv'), 'w', encoding='utf-8') as f:
f.write(target)
create_binary_files(csv_dir, bin_dir, do_il=True, do_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)))
self.assertEqual(len(files), len(glob.glob(os.path.join(bin_dir, 'RI_2{}*.bin'.format(os.sep)))))
for filename in (f + '.bin' for f in files):
self.assertTrue(os.path.exists(os.path.join(bin_dir, 'RI_2', filename)))
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')
create_binary_files(csv_dir, bin_dir)
self.assertEqual(0, len(glob.glob(os.path.join(csv_dir, '*.bin'))))
def test_contains_il_and_standard_files_but_do_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)
create_binary_files(csv_dir, bin_dir, do_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,
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.create_binary_files(ri_layers[idx]['directory'],
ri_layers[idx]['directory'],
do_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))
