def test_untransform(self): nc = NomCat() nc.train(self.m) orig = self.m.data[0] trans = nc.transform(orig) self.assertTrue((orig == nc.untransform(trans)).all()) orig = self.m.data[1] trans = nc.transform(orig) self.assertTrue((orig == nc.untransform(trans)).all()) orig = self.m.data[2] trans = nc.transform(orig) self.assertTrue((orig == nc.untransform(trans)).all())
def test_transform(self): nc = NomCat() nc.train(self.m) self.assertTrue(([10, 0, 1, 0] == nc.transform(self.m.data[0])).all()) self.assertTrue(([20, 0, 0, 1] == nc.transform(self.m.data[1])).all()) self.assertTrue(([30, 1, 0, 0] == nc.transform(self.m.data[2])).all())