added defined method test.

2 files changed, 13 insertions, 6 deletions

diff --git a/test_evaluate_expression.py b/test_evaluate_expression.py
index 171e6ed..2fc62cc 100644
--- a/test_evaluate_expression.py
+++ b/test_evaluate_expression.py
@@ -6,22 +6,26 @@ class TestEvaluateExpression(unittest.TestCase):
 
 	@classmethod
 	def setUpClass(cls):
-		cls.src_fil_r = dialoguer.Dialogue('test_evaluate_expression.r')
+		pass
 
 	@classmethod
 	def tearDownClass(cls):
-		cls.src_fil_r.close()
+		pass
 
 	def setUp(self):
-		pass
+		self.src_fil_r = dialoguer.Dialogue('test_evaluate_expression.r')
 
 	def tearDown(self):
-		pass
+		self.src_fil_r.close()
 
-	def test_evaluate_expression(self):
+	def test_built_in_method(self):
 		res_1 = self.src_fil_r.evaluate_expression('sum', 1, 2)
 		self.assertEqual(res_1, 3)
 
+	def test_defined_method(self):
+		res_1 = self.src_fil_r.evaluate_expression('test_method', 1, 2)
+		self.assertEqual(res_1, 3)
+
 
 if __name__ == '__main__':
 	unittest.main()
diff --git a/test_evaluate_expression.r b/test_evaluate_expression.r
index d5b533b..cf60a67 100644
--- a/test_evaluate_expression.r
+++ b/test_evaluate_expression.r
@@ -1 +1,4 @@
-i <- 1
\ No newline at end of file
+test_method <- function (arg_1, arg_2) {
+	result <- arg_1 + arg_2
+	return(result)
+}
\ No newline at end of file