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

@@ -1,16 +1,15 @@
 require File.expand_path("../test_helper", File.dirname(__FILE__))
+require "rails_erd/diagram/graphviz"
 
 class GraphvizTest < ActiveSupport::TestCase
   def setup
     RailsERD.options.filetype = :png
     RailsERD.options.warn = false
-    load "rails_erd/diagram/graphviz.rb"
   end
 
   def teardown
     FileUtils.rm Dir["erd*.*"] rescue nil
-    RailsERD::Diagram.send :remove_const, :Graphviz rescue nil
   end
 
   def diagram(options = {})
     @diagram ||= Diagram::Graphviz.new(Domain.generate(options), options).tap do |diagram|
       diagram.generate