Added image display option in testing

alishdipani · alishdipani · commit 19b77fe5638f · 2020-03-13T12:56:38.000+05:30
diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
@@ -7,6 +7,7 @@
 SPEC_ROOT = File.dirname(__FILE__) + "/"
 TEMP_DIR = SPEC_ROOT + "temp/"
 FIXTURES_DIR = SPEC_ROOT + "fixtures/"
+display_img = nil
 
 backend = ENV['RUBYPLOT_BACKEND']
 
@@ -50,9 +51,21 @@ def compare_with_reference(test_image, reference_image)
   config.before(:suite) do
     FileUtils.mkdir_p TEMP_DIR
     FileUtils.mkdir_p FIXTURES_DIR
+    $stdout.puts 'Do you want to display the image? [y/n]'
+    display_img = $stdin.gets.chomp.strip[0].downcase
+    while display_img != "y" && display_img != "n"
+      $stdout.puts 'Please enter y or n'
+      display_img = $stdin.gets.chomp.strip[0].downcase
+    end
+    if display_img == "y"
+      display_img = true
+    else
+      display_img = false
+    end
   end
-  
-  config.after(:example) do |example| 
+
+  config.after(:example) do |example|
+    @figure.show if display_img
     if @figure.is_a?(Rubyplot::Artist::Figure)
       plot_name = example.description.split.join("_") + ".png"
       base_image = TEMP_DIR + plot_name
