def test_cont_lists(self):
contact = Contact()
lists = [{'id': '123'}, {'id': '456'}, {'id': '789'}]
contact_list = Contact_List({'id': '321'})
contact.set_lists(lists)
self.assertEqual(len(contact.get_lists()), 3)
self.assertEqual(contact.get_list(2).get_id(), '789')
contact.add_list(contact_list)
contact.add_list_id('654')
contact.remove_list_id('123')
contact.remove_list(1)
self.assertEqual(contact.get_list(0).get_id(), '456')
self.assertEqual(contact.get_list(2).get_id(), '654')
self.assertEqual(len(contact.get_lists()), 3)
contact.remove_all_lists()
self.assertEqual(contact.get_lists(), None)
def test_cont_lists(self):
contact = Contact()
lists = [{"id": "123"}, {"id": "456"}, {"id": "789"}]
contact_list = Contact_List({"id": "321"})
contact.set_lists(lists)
self.assertEqual(len(contact.get_lists()), 3)
self.assertEqual(contact.get_list(2).get_id(), "789")
contact.add_list(contact_list)
contact.add_list_id("654")
contact.remove_list_id("123")
contact.remove_list(1)
self.assertEqual(contact.get_list(0).get_id(), "456")
self.assertEqual(contact.get_list(2).get_id(), "654")
self.assertEqual(len(contact.get_lists()), 3)
contact.remove_all_lists()
self.assertEqual(contact.get_lists(), None)
