test/stats_test.rb in fastly-1.01 vs test/stats_test.rb in fastly-1.1.0

@@ -1,14 +1,10 @@
-#!/usr/bin/env ruby
+require 'helper'
 
-require 'rubygems'
-require 'test/unit'
-require File.expand_path(File.dirname(__FILE__) + '/helper')
-
 FROM = "2011-01-01 00:00:00"
 
-class StatsTest < Test::Unit::TestCase
+class StatsTest < Fastly::TestCase
 
   def setup
     opts = login_opts(:api_key).merge(:use_curb => false)
     begin
       @fastly = Fastly.new(opts)
@@ -84,6 +80,6 @@
     rescue Fastly::Error => e
     end
     assert stats.nil?
   end
 
-end
\ No newline at end of file
+end