def test_to_dict(self): item_link = ItemLink(type=self.link_type, item_id=self.item_1.id, linked_id=self.item_2.id) db.session.add(item_link) db.session.flush() self.assertDictEqual(item_link.to_dict(), {'type': item_link.type, 'item_id': item_link.item_id, 'linked_id': item_link.linked_id})