Python OasisLookupFactory.get_keys_base Examples

Example #1

File: test_lookup.py Project: zejiang-unsw/OasisLMF

 def test_entries_are_dictionaries_success_only_is_false___all_entries_are_included(
         self, data):
     with patch(
             'oasislmf.model_preparation.lookup.OasisLookupFactory.get_exposure'
     ):
         self.create_fake_lookup(return_value=data)
         mock_df = pd.DataFrame.from_dict(data)
         res = list(
             olf.get_keys_base(lookup=self.lookup_instance,
                               loc_df=mock_df,
                               success_only=False))
         self.assertEqual(res, data)

Example #2

 def test_entries_are_dictionaries_success_only_is_false___all_entries_are_included(self, data):
     self.create_fake_lookup(return_value=data)
     mock_df = pd.DataFrame.from_dict(data)
     res = list(olf.get_keys_base(lookup=self.lookup_instance, loc_df=mock_df, success_only=False))
     self.assertEqual(res, data)

Example #3

 def test_entries_are_dictionaries_success_only_is_true___only_successes_are_included(self, data):
     self.create_fake_lookup(return_value=data)
     mock_df = pd.DataFrame.from_dict(data)
     res = list(olf.get_keys_base(lookup=self.lookup_instance, loc_df=mock_df, success_only=True))
     self.assertEqual(res, [d for d in data if d['status'] == 'success'])