Replaced custom assert_num_queries with the builtin assertNumQueries.