From d107de488121be43306b0cda65d009f99af775ba Mon Sep 17 00:00:00 2001 From: TinWoodman92 Date: Mon, 19 Jun 2023 22:15:57 -0500 Subject: updated test import to include integers --- test_import_variable.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'test_import_variable.py') 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() -- cgit v1.2.3