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()