def test_exposure_string_is_provided___file_content_is_loaded(self, data):
columns = loc_required_cols
exposure_str = pd.DataFrame(columns=columns, data=data).to_csv(index=False)
res_str = olf.get_exposure(source_exposure=exposure_str).to_csv(index=False)
self.assertEqual(exposure_str, res_str)
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 = olf.get_exposure(source_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 = olf.get_exposure(source_exposure_fp=f.name).to_csv(index=False)
self.assertEqual(exposure_str, res_str)
def test_file_is_provided___file_content_is_loaded(self, data):
columns = loc_required_cols
exposure_str = pd.DataFrame(columns=columns, data=data).to_csv(index=False)
f = NamedTemporaryFile('w', delete=False)
try:
f.writelines(exposure_str)
f.close()
res_str = olf.get_exposure(source_exposure_fp=f.name).to_csv(index=False)
self.assertEqual(exposure_str, res_str)
finally:
os.remove(f.name)
def test_no_file_or_exposure_are_provided___oasis_exception_is_raised(
self):
with self.assertRaises(OasisException):
olf.get_exposure()