Changeset 10288:e9bfacf8309b in orange


Ignore:
Timestamp:
02/17/12 14:32:53 (2 years ago)
Author:
anzeh <anze.staric@…>
Branch:
default
rebase_source:
3967c5b8360008419e3bdf3e6f89e901dca34069
Message:

Added tests for ConfusionMatrix

File:
1 edited

Legend:

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

    r10280 r10288  
    123123        pt = testing.proportion_test([self.learner], ds, times=1) 
    124124        self.assertEqual(pt.number_of_iterations, 1) 
    125         print pt.number_of_iterations 
    126125        ca = scoring.CA(pt) 
    127126        self.assertEqual(len(ca), 1) 
     
    147146        self.assertEqual(len(ca), 1) 
    148147 
     148 
     149class TestConfusionMatrix(unittest.TestCase): 
     150    def test_construct_confusion_matrix_from_multiclass(self): 
     151        learner = random_learner 
     152        ds = data.Table("iris") 
     153        pt = testing.proportion_test([learner], ds, times=1) 
     154        cm = scoring.confusion_matrices(pt) 
     155 
     156        self.assertTrue(isinstance(cm[0], list)) 
     157 
     158 
     159    def test_construct_confusion_matrix_from_biclass(self): 
     160        learner = random_learner 
     161        ds = data.Table("monks-1") 
     162        pt = testing.proportion_test([learner], ds, times=1) 
     163        cm = scoring.confusion_matrices(pt, class_index=1) 
     164 
     165        self.assertTrue(hasattr(cm[0], "TP")) 
     166 
     167 
     168class TestConfusionMatrix(unittest.TestCase): 
     169    def test_construct_confusion_matrix_from_multiclass(self): 
     170        learner = random_learner 
     171        ds = data.Table("iris") 
     172        pt = testing.proportion_test([learner], ds, times=1) 
     173        cm = scoring.confusion_matrices(pt) 
     174 
     175        self.assertTrue(isinstance(cm[0], list)) 
     176 
     177 
     178    def test_construct_confusion_matrix_from_biclass(self): 
     179        learner = random_learner 
     180        ds = data.Table("monks-1") 
     181        pt = testing.proportion_test([learner], ds, times=1) 
     182        cm = scoring.confusion_matrices(pt, class_index=1) 
     183 
     184        self.assertTrue(hasattr(cm[0], "TP")) 
     185 
    149186if __name__ == '__main__': 
    150187    unittest.main() 
Note: See TracChangeset for help on using the changeset viewer.