updated test import to include integers

1 files changed, 7 insertions, 1 deletions

diff --git a/test_import_variable.py b/test_import_variable.py
index ca3cba4..4fa45e9 100644
--- a/test_import_variable.py
+++ b/test_import_variable.py
@@ -19,12 +19,18 @@ class TestImportVariable(unittest.TestCase):
 	def tearDown(self):
 		self.src_fil_r.close()
 
-	def test_import_variable(self):
+	def test_import_string(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, 'Initializing Client - R')
 		self.assertEqual(msg_2, 'Initialized Client - R')
 
+	def test_import_integer(self):
+		int_1 = self.src_fil_r.import_variable('int_1')
+		int_2 = self.src_fil_r.import_variable('int_2')
+		self.assertEqual(int_1, 3)
+		self.assertEqual(int_2, 155)
+
 
 if __name__ == '__main__':
 	unittest.main()