lib/assert/runner.rb in assert-0.1.0 vs lib/assert/runner.rb in assert-0.2.0

@@ -10,18 +10,14 @@
       @suite = suite
       @view = view
     end
 
     def run(*args)
-      @suite.setup_blocks.each do |setup| # TODO: tests!
-        setup.call
-      end
+      @suite.setup
       @view.render do
         benchmark { run_suite }
       end
-      @suite.teardown_blocks.each do |teardown| # TODO: tests!
-        teardown.call
-      end
+      @suite.teardown
       count(:failed) + count(:errored)
     end
 
     def count(type)
       @suite.count(type)