test module for KE examples added

dream/tests/testKEExamples.py

+# ===========================================================================
+# Copyright 2014 Nexedi SA
+#
+# This file is part of DREAM.
+#
+# DREAM is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# DREAM is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with DREAM.  If not, see <http://www.gnu.org/licenses/>.
+# ===========================================================================
+
+from unittest import TestCase
+import os
+project_path = os.path.split(os.path.split(os.path.split(__file__)[0])[0])[0]
+import xlrd
+import json
+import glob
+
+
+class KnowledgeExtractionExamples(TestCase):
+    """
+    Test KE examples
+    """
+    def testTwoParallelStations(self):
+        from dream.KnowledgeExtraction.KEtool_examples.TwoParallelStations.TwoParallelStations_example import main 
+        filepath=glob.glob(os.path.join(project_path, "dream", "KnowledgeExtraction", "KEtool_examples", 
+                                                        "TwoParallelStations"))[0]
+        workbook = xlrd.open_workbook(filepath+'\\inputData.xls')
+        jsonFile = open(filepath+'\\JSON_TwoParallelStations.json')      
+        result = main(test=1,workbook=workbook,jsonFile=jsonFile)
+        result_data = json.loads(json.loads(result))
+        
+        result_data=result_data['result']['result_list'][0]
+        elementList=result_data.get('elementList',[])
+        for element in elementList:
+            if element['id']=='E1':
+                self.assertEquals(element['results']['throughput'][0], 86)
+            if element['id']=='St1':
+                self.assertTrue(99.93<element['results']['working_ratio'][0]<99.94)
+        jsonFile.close()
+     
+   
\ No newline at end of file