def test_key_val_parsing(self):
"""
Test parsing of keyval strings
"""
self.assertEqual(get_value_from_input('fruits="pear,kiwi,banana"'),
{'fruits': '"pear,kiwi,banana"'})
self.assertEqual(get_value_from_input('berries='), {'berries': ''})
self.assertEqual(get_value_from_input('pythagoras=a2+b2=c2'),
{'pythagoras': 'a2+b2=c2'})
def test_environ_extraction(self):
"""
Test inclusion of variables from the environ
"""
os.environ['vegetables'] = '"carrot,peas,green beans"'
os.environ['smellythings'] = ''
self.assertEqual(get_value_from_input('vegetables'),
{'vegetables': '"carrot,peas,green beans"'})
self.assertEqual(get_value_from_input('smellythings'),
{'smellythings': ''})
def test_environ_extraction(self):
"""
Test inclusion of variables from the environ
"""
os.environ['vegetables'] = '"carrot,peas,green beans"'
os.environ['smellythings'] = ''
self.assertEqual(
get_value_from_input('vegetables'),
{'vegetables': '"carrot,peas,green beans"'}
)
self.assertEqual(
get_value_from_input('smellythings'),
{'smellythings': ''}
)
def test_key_val_parsing(self):
"""
Test parsing of keyval strings
"""
self.assertEqual(
get_value_from_input('fruits="pear,kiwi,banana"'),
{'fruits': '"pear,kiwi,banana"'}
)
self.assertEqual(
get_value_from_input('berries='),
{'berries': ''}
)
self.assertEqual(
get_value_from_input('pythagoras=a2+b2=c2'),
{'pythagoras': 'a2+b2=c2'}
)
def test_combination(self):
# Test having both environment vars and arbitrary keyvals
os.environ['bears'] = 'polar,brown,black'
os.environ['notbears'] = 'unicorn,griffin,sphinx,otter'
command_line = ['bears', 'notbears', 'e=mc2', 'v=iR', 'qwertyuiop']
expected = {
'bears': 'polar,brown,black',
'notbears': 'unicorn,griffin,sphinx,otter',
'e': 'mc2',
'v': 'iR',
}
result = {}
for variable in command_line:
result.update(get_value_from_input(variable))
self.assertDictEqual(result, expected)
def test_combination(self):
"""
Test having both environment vars and arbitrary keyvals
"""
os.environ['bears'] = 'polar,brown,black'
os.environ['notbears'] = 'unicorn,griffin,sphinx,otter'
command_line = ['bears', 'notbears', 'e=mc2', 'v=iR', 'qwertyuiop']
expected = {
'bears': 'polar,brown,black',
'notbears': 'unicorn,griffin,sphinx,otter',
'e': 'mc2',
'v': 'iR',
}
result = {}
for variable in command_line:
result.update(get_value_from_input(variable))
self.assertDictEqual(result, expected)
