def test_prettify_address():
from Panorama import prettify_address
address = {
'@name': 'my_name',
'ip-netmask': '1.1.1.1',
'description': 'lala'
}
response = prettify_address(address)
expected = {
'Name': 'my_name',
'IP_Netmask': '1.1.1.1',
'Description': 'lala'
}
assert response == expected
def test_prettify_address_tag_none():
from Panorama import prettify_address
address = {'@name': 'test', 'ip-netmask': '1.1.1.1', 'tag': None}
response = prettify_address(address)
expected = {'Name': 'test', 'IP_Netmask': '1.1.1.1'}
assert response == expected
