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)
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)
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'])