added defined method test.

1 files changed, 9 insertions, 5 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()