def test_only_per_feature_override(self):
m = example_weight_map.ExampleWeightMap(
per_feature_override={
types.FeaturePath(['foo']): 'w1',
})
self.assertEqual('w1', m.get(types.FeaturePath(['foo'])))
self.assertIsNone(m.get(types.FeaturePath(['feature'])))
self.assertEqual(m.all_weight_features(), frozenset(['w1']))
def test_per_feature_override(self):
m = example_weight_map.ExampleWeightMap(weight_feature='w',
per_feature_override={
types.FeaturePath(['foo']):
'w1',
types.FeaturePath(['bar']):
'w2'
})
self.assertEqual('w1', m.get(types.FeaturePath(['foo'])))
self.assertEqual('w2', m.get(types.FeaturePath(['bar'])))
self.assertEqual('w', m.get(types.FeaturePath(['feature'])))
self.assertEqual(m.all_weight_features(), frozenset(['w', 'w1', 'w2']))
def example_weight_map(self):
return example_weight_map.ExampleWeightMap(
self.weight_feature, self._per_feature_weight_override)
def test_only_global_weight_feature(self):
m = example_weight_map.ExampleWeightMap(weight_feature='w')
self.assertEqual(m.get(types.FeaturePath(['feature'])), 'w')
self.assertEqual(m.all_weight_features(), frozenset(['w']))
def test_no_weight_feature(self):
m = example_weight_map.ExampleWeightMap()
self.assertIsNone(m.get(types.FeaturePath(['feature'])))
self.assertEmpty(m.all_weight_features())
