fix complex_number_arithmetic with 0 bug

[mint/mint-lib.git] / lib / mint / generator / complex_number_arithmetic.rb

diff --git a/lib/mint/generator/complex_number_arithmetic.rb b/lib/mint/generator/complex_number_arithmetic.rb
index 6991f03..2089d7e 100644 (file)
--- a/lib/mint/generator/complex_number_arithmetic.rb
+++ b/lib/mint/generator/complex_number_arithmetic.rb
@@ -45,7 +45,7 @@ module Mint::Generator
       real_part      = create_integer(options[:min], options[:max], false)
       imaginary_part = create_integer(options[:min], options[:max], false)
       return create_integer(2, 9, false).to_s if [real_part, imaginary_part].include? 0
-      "(#{real_part} #{plus_or_minus} #{imaginary_part.to_s.sub('1', '')}%i)"
+      "(#{real_part} #{plus_or_minus} #{imaginary_part.to_s.sub(/\A1\z/, '')}%i)"
     end
 
     def plus_or_minus