class Test(unittest.TestCase):
def setUp(self):
self.datafile = '../data'
self.ds = DataSource(self.datafile)
def testInitialization(self):
ds = DataSource('../data')
self.assertEqual(ds.targetAttr, 'Purchase?')
self.assertEqual(len(ds.dataset), 20)
self.assertEqual(ds.datafile, self.datafile)
def testInitialization2(self):
ds = DataSource('../data', 'Purchase?')
self.assertEqual(ds.targetAttr, 'Purchase?')
self.assertEqual(len(ds.dataset), 20)
self.assertEqual(ds.datafile, self.datafile)
def testMojorityValue(self):
self.assertEqual(self.ds.majorityValue(self.ds.dataset, 'Purchase?'),'will buy')
def testDualization(self):
self.ds.dualization('../data.o')
def tearDown(self):
del self.ds
