1 files changed, 47 insertions, 0 deletions
diff --git a/yaksh/tests/test_scilab_evaluation.py b/yaksh/tests/test_scilab_evaluation.py
new file mode 100644
index 0000000..30af041
--- /dev/null
+++ b/yaksh/tests/test_scilab_evaluation.py
@@ -0,0 +1,47 @@
+import unittest
+import os
+from yaksh.scilab_code_evaluator import ScilabCodeEvaluator
+from yaksh.settings import SERVER_TIMEOUT
+
+class ScilabEvaluationTestCases(unittest.TestCase):
+    def setUp(self):
+        self.language = "scilab"
+        self.ref_code_path = "scilab_files/test_add.sce"
+        self.in_dir = "/tmp"
+        self.test_case_data = []
+        self.timeout_msg = ("Code took more than {0} seconds to run. "
+                            "You probably have an infinite loop in your code.").format(SERVER_TIMEOUT)
+        self.test = None
+
+    def test_correct_answer(self):
+        user_answer = "funcprot(0)\nfunction[c]=add(a,b)\n\tc=a+b;\nendfunction"
+        get_class = ScilabCodeEvaluator(self.test_case_data, self.test,
+                                             self.language, user_answer,
+                                             self.ref_code_path, self.in_dir)
+        result = get_class.evaluate()
+
+        self.assertTrue(result.get("success"))
+        self.assertEqual(result.get("error"), "Correct answer")
+
+    def test_error(self):
+        user_answer = "funcprot(0)\nfunction[c]=add(a,b)\n\tc=a+b;\ndis(\tendfunction"
+        get_class = ScilabCodeEvaluator(self.test_case_data, self.test,
+                                             self.language, user_answer,
+                                             self.ref_code_path, self.in_dir)
+        result = get_class.evaluate()
+ 
+        self.assertFalse(result.get("success"))
+        self.assertTrue("error" in result.get("error"))
+
+    def test_infinite_loop(self):
+        user_answer = "funcprot(0)\nfunction[c]=add(a,b)\n\tc=a;\nwhile(1==1)\nend\nendfunction"
+        get_class = ScilabCodeEvaluator(self.test_case_data, self.test,
+                                         self.language, user_answer,
+                                         self.ref_code_path, self.in_dir)
+        result = get_class.evaluate()
+
+        self.assertFalse(result.get("success"))
+        self.assertEquals(result.get("error"), self.timeout_msg)
+
+if __name__ == '__main__':
+    unittest.main()