c24fdc1414900a4172c996cdc2459a494322ba2e,tests/test_transerror.py,ErrorsTests,testConfusionMatrix,#ErrorsTests#,26

Before Change

        self.failUnlessRaises(ZeroDivisionError, lambda x:x.percentCorrect, cm)
        No samples -- raise exception

        cm.add(reg, N.array([1,2,1,2,2,2,3,2,1]))

        self.failUnlessEqual(len(cm.sets), 1,
            msg="Should have a single set so far")
        self.failUnlessEqual(cm.matrix.shape, (3,3),

After Change

    def testConfusionMatrix(self):
        data = N.array([1,2,1,2,2,2,3,2,1], ndmin=2).T
        reg = [1,1,1,2,2,2,3,3,3]
        regl = [1,2,1,2,2,2,3,2,1]

        // Check if we are ok with any input type - either list, or N.array, or tuple
        for t in [reg, tuple(reg), list(reg), N.array(reg)]:
            for p in [regl, tuple(regl), list(regl), N.array(regl)]:
                cm = ConfusionMatrix(targets=t, predictions=p)
                // check table content
                self.failUnless((cm.matrix == [[2,1,0],[0,3,0],[1,1,1]]).all())

        // Do a bit more thorough checking
        cm = ConfusionMatrix()
        self.failUnlessRaises(ZeroDivisionError, lambda x:x.percentCorrect, cm)
        No samples -- raise exception

Instances