def test_search_for(self): query = Query() query.search_for(GNC_ID_INVOICE) self.assertEqual(query.get_search_for(), GNC_ID_INVOICE) obj_type = 'gncInvoice' query.search_for(obj_type) self.assertEqual(query.get_search_for(), obj_type)