def test_cont_address(self):
        contact = Contact()
        address = Address()

        address.set_state("Oregon")
        address.set_city("Gravity Falls")
        address.set_address_type("BUSINESS")

        self.assertEqual(contact.get_addresses(), None)

        contact.set_address(address)

        self.assertEqual(contact.get_address("BUSINESS").get_city(), "Gravity Falls")

        contact.remove_address("PERSONAL")

        self.assertEqual(contact.get_address("BUSINESS").get_city(), "Gravity Falls")

        contact.clear_addresses()

        self.assertEqual(contact.get_address("BUSINESS"), None)
    def test_cont_address(self):
        contact = Contact()
        address = Address()

        address.set_state('Oregon')
        address.set_city('Gravity Falls')
        address.set_address_type('BUSINESS')

        self.assertEqual(contact.get_addresses(), None)

        contact.set_address(address)

        self.assertEqual(
            contact.get_address('BUSINESS').get_city(), 'Gravity Falls')

        contact.remove_address('PERSONAL')

        self.assertEqual(
            contact.get_address('BUSINESS').get_city(), 'Gravity Falls')

        contact.clear_addresses()

        self.assertEqual(contact.get_address('BUSINESS'), None)