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