How to test a data retrieval module?

I am working on a project and have a data retrieval module written on python. All it does is build a query using string manipulations and then executes the query using another module (a connection module). I want to make unit tests properly. Does anyone have any recommendations on techniques? Is mocking a proper way to do so?