Ignore:
Timestamp:
08/02/13 16:10:29 (9 months ago)
Author:
Ales Erjavec <ales.erjavec@…>
Branch:
default
Message:

Preserve the domain's meta attributes and class_vars.

File:
1 edited

Legend:

Unmodified
Added
Removed
  • Orange/testing/unit/tests/test_feature_selection.py

    r11646 r11647  
    5151        self.assertEqual(above, [a.name for a in new_data.domain.attributes]) 
    5252        self.assertEqual(new_data.domain.class_var, self.data.domain.class_var) 
    53          
    54          
    55          
    56          
    57          
    58      
     53 
     54    def test_select_features_subset(self): 
     55        data = Orange.data.Table("lenses") 
     56 
     57        d1 = selection._select_features_subset(data, []) 
     58        self.assertSequenceEqual(d1.domain.features, []) 
     59        self.assertIs(d1.domain.class_var, data.domain.class_var) 
     60 
     61        d1 = selection._select_features_subset(data, [data.domain[0]]) 
     62        self.assertSequenceEqual(d1.domain.features, [data.domain[0]]) 
     63        self.assertIs(d1.domain.class_var, data.domain.class_var) 
     64 
     65        domain = Orange.data.Domain(data.domain.features[:2], 
     66                                    data.domain.class_var, 
     67                                    class_vars=[data.domain.features[2]]) 
     68        domain.add_metas({-1, data.domain.features[3]}) 
     69        data = Orange.data.Table(domain, data) 
     70 
     71        d1 = selection._select_features_subset(data, [data.domain[0]]) 
     72        self.assertSequenceEqual(d1.domain.features, [data.domain[0]]) 
     73        self.assertIs(d1.domain.class_var, data.domain.class_var) 
     74        self.assertSequenceEqual(d1.domain.class_vars, data.domain.class_vars) 
     75        self.assertEqual(d1.domain.get_metas(), data.domain.get_metas()) 
     76 
    5977if __name__ == "__main__": 
    6078    unittest.main() 
    61      
Note: See TracChangeset for help on using the changeset viewer.