test/unit/diagram_test.rb in rails-erd-1.2.2 vs test/unit/diagram_test.rb in rails-erd-1.3.0

@@ -1,15 +1,8 @@
 require File.expand_path("../test_helper", File.dirname(__FILE__))
+require "rails_erd/diagram"
 
 class DiagramTest < ActiveSupport::TestCase
-  def setup
-    load "rails_erd/diagram.rb"
-  end
-
-  def teardown
-    RailsERD.send :remove_const, :Diagram
-  end
-
   def retrieve_entities(options = {})
     klass = Class.new(Diagram)
     [].tap do |entities|
       klass.class_eval do
         each_entity do |entity, attributes|