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