Set up `rake test` task