def test__replace_vars_with_None(self): vars = {'__INPUT_FILE__': None} args = { 'file.txt': '__INPUT_FILE__.txt', } replaced = config_loader._replace_vars(args, vars) self.assertEqual('__INPUT_FILE__.txt', replaced.get('file.txt'))
def test__replace_vars(self): vars = {'__INPUT_FILE__': '/tmp/video.mp4'} args = {'bool': True, 'num': 123, 'str': 'Hello', 'file.txt': '__INPUT_FILE__.txt', 'message': 'This is __INPUT_FILE__.', } replaced = config_loader._replace_vars(args, vars) self.assertTrue(replaced.get('bool')) self.assertEqual(123, replaced.get('num')) self.assertEqual('Hello', replaced.get('str')) self.assertEqual('/tmp/video.mp4.txt', replaced.get('file.txt')) self.assertEqual('This is /tmp/video.mp4.', replaced.get('message'))