class TestEchoTreeEval(unittest.TestCase): def setUp(self): currDir = os.path.dirname(os.path.realpath(__file__)); self.dbFileName = os.path.join(currDir, "../../Resources/henryBlog.db"); self.evaluator = Evaluator(self.dbFileName); self.tokenFile = os.path.join(currDir, "henry_Tokens.txt"); self.testArity = 2; @SkipTest def test_bigrams(self): self.testArity = 2; perfNum = self.evaluator.measurePerformance("/tmp/echoTreeEvalTestBigrams.csv", self.dbFileName, self.testArity, [self.tokenFile], verbosity=Verbosity.DEBUG ); self.assertEqual(0.373015873015873, perfNum); # Check the output file: with open("/tmp/echoTreeEvalTestBigrams.csv") as fd: spreadsheet = fd.readlines(); self.assertEquals('EmailID,SentenceID,SentenceLen,Failures,OutofSeq,InputSavings,Depth_1,Depth_2,DepthWeightedScore\n', spreadsheet[0]); self.assertEquals('10639,0,10,4,0,25.0,2,3,0.388888888889\n', spreadsheet[1]); self.assertEquals('10639,1,15,9,0,16.6666666667,5,0,0.357142857143\n', spreadsheet[2]); def test_trigrams(self): self.testArity = 3; perfNum = self.evaluator.measurePerformance("/tmp/echoTreeEvalTestTrigrams.csv", self.dbFileName, self.testArity, [self.tokenFile], verbosity=Verbosity.DEBUG ); self.assertEqual(0.30753968253968256, perfNum); # Check the output file: with open("/tmp/echoTreeEvalTestTrigrams.csv") as fd: spreadsheet = fd.readlines(); self.assertEquals('EmailID,SentenceID,SentenceLen,Failures,OutofSeq,InputSavings,Depth_1,Depth_2,DepthWeightedScore\n', spreadsheet[0]); self.assertEquals('10639,0,10,6,0,14.2857142857,1,2,0.222222222222\n', spreadsheet[1]); self.assertEquals('10639,1,15,8,0,18.8888888889,5,1,0.392857142857\n', spreadsheet[2]);
def setUp(self): currDir = os.path.dirname(os.path.realpath(__file__)); self.dbFileName = os.path.join(currDir, "../../Resources/henryBlog.db"); self.evaluator = Evaluator(self.dbFileName); self.tokenFile = os.path.join(currDir, "henry_Tokens.txt"); self.testArity = 2;