1 files changed, 30 insertions, 0 deletions
diff --git a/test_import_variable.py b/test_import_variable.py
new file mode 100644
index 0000000..c718e92
--- /dev/null
+++ b/test_import_variable.py
@@ -0,0 +1,30 @@
+import unittest
+import dialoguer
+
+
+class TestEmployee(unittest.TestCase):
+
+	@classmethod
+	def setUpClass(cls):
+		pass
+
+	@classmethod
+	def tearDownClass(cls):
+		pass
+
+	def setUp(self):
+		self.src_fil_r = dialoguer.Dialogue('test_import_variable.r')
+		self.src_fil_r.open()
+	
+	def tearDown(self):
+		self.src_fil_r.close()
+
+	def test_import_variable(self):
+		msg_1 = self.src_fil_r.import_variable('msg_1')
+		msg_2 = self.src_fil_r.import_variable('msg_2')
+		self.assertEqual(msg_1, b'Initializing Client - R')
+		self.assertEqual(msg_2, b'Initialized Client - R')
+
+
+if __name__ == '__main__':
+	unittest.main()