Query classifier tests

Tests for the query classifier