def test_exposure_string_is_provided___file_content_is_loaded(self, data):
columns = ['first', 'second']
exposure_str = _unicode(
pd.DataFrame(columns=columns, data=data).to_csv(index=False))
res_str = OasisLookupFactory.get_model_exposure(
model_exposure=exposure_str).to_csv(index=False)
self.assertEqual(exposure_str, res_str)
def test_file_is_provided___file_content_is_loaded(self, data):
columns = ['first', 'second']
exposure_str = _unicode(
pd.DataFrame(columns=columns, data=data).to_csv(index=False))
with NamedTemporaryFile('w') as f:
f.writelines(exposure_str)
f.flush()
res_str = OasisLookupFactory.get_model_exposure(
model_exposure_fp=f.name).to_csv(index=False)
self.assertEqual(exposure_str, res_str)
def test_no_file_or_exposure_are_provided___oasis_exception_is_raised(
self):
with self.assertRaises(OasisException):
OasisLookupFactory.get_model_exposure()