diff options
| -rw-r--r-- | test_import_variable.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test_import_variable.py b/test_import_variable.py index c718e92..ca3cba4 100644 --- a/test_import_variable.py +++ b/test_import_variable.py @@ -2,7 +2,7 @@ import unittest import dialoguer -class TestEmployee(unittest.TestCase): +class TestImportVariable(unittest.TestCase): @classmethod def setUpClass(cls): @@ -22,8 +22,8 @@ class TestEmployee(unittest.TestCase): 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') + self.assertEqual(msg_1, 'Initializing Client - R') + self.assertEqual(msg_2, 'Initialized Client - R') if __name__ == '__main__': |