Fix setup in scatter_plot test

red-data-tools · May 22, 2024 · 3bed5b6 · 3bed5b6
1 parent 048c9a5
commit 3bed5b6
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/test/plot_methods/scatter_plot_test.rb b/test/plot_methods/scatter_plot_test.rb
@@ -3,11 +3,12 @@ class PlotMethodScatterPlotTest < Test::Unit::TestCase
 
   sub_test_case("function-call style") do
     def setup
-      super
-
-      @x = Array.new(20) {|i| rand(i.to_f) }
-      @y = Array.new(20) {|i| rand * @x[i] }
-      @data = {x: @x, y: @y}
+      super do
+        @x = Array.new(20) {|i| rand(i.to_f) }
+        @y = Array.new(20) {|i| rand * @x[i] }
+        @data = {x: @x, y: @y}
+        yield
+      end
     end
 
     sub_test_case("wide-form input") do