Clarify the Output type a little

2025-09-17 09:08:13 +08:00 · 2021-05-29 16:17:59 +09:00
parent 2eca86268c
commit f824135653
3 changed files with 5 additions and 3 deletions
--- a/lib/youplot/command.rb
+++ b/lib/youplot/command.rb
@@ -182,9 +182,9 @@ module YouPlot

    def output_plot(plot)
      case options[:output]
-      when IO
+      when IO, StringIO
        plot.render(options[:output])
-      else
+      when String, Tempfile
        File.open(options[:output], 'w') do |f|
          plot.render(f)
        end
@@ -193,7 +193,7 @@ module YouPlot

    def output_plot_progressive(plot)
      case options[:output]
-      when IO
+      when IO, StringIO
        # RefactorMe
        out = StringIO.new(String.new)
        def out.tty?