def test_parser(self): fname = "network_dataset.csv" rows = 0 with open(fname, 'rb') as f: dataFromFile = reader(f) for row in dataFromFile: rows += 1 self.assertEqual(rows, len(parse_file(fname)))
pass def test_init(self): pass #------------------------------------------------------------------------------- class TestKMeans(unittest.TestKMeans): def setUp(self): pass def test_kmean(self): pass #------------------------------------------------------------------------------ def run_suite(suite): tests = unittest.TestLoader().loadTestsFromTestCase(suite) return unittest.TextTestRunner(verbosity=2).run(tests) #------------------------------------------------------------------------------ if __name__ == '__main__': result = run_suite(TestParser) if !result.wasSuccessful(): print "" print "Now loading data..." item_list = parse_file("network_dataset.csv") print "Data loaded" print "" run_suite(TestUtil) print "Now testing Analysis..." run_suite(TestAnalysis)