Added some tests for basic querying

Models added to session seem to be removed after a get_expect call.
Got brief, non-permanent solution