Fix comparison with Float::INFINITY

Fixes [Bug #17945]
ruby · Jun 15, 2021 · 953d907 · 953d907
1 parent f3c4836
commit 953d907
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/lib/date.rb b/lib/date.rb
@@ -30,6 +30,8 @@ def +@() self.class.new(+d) end
     def <=>(other)
       case other
       when Infinity; return d <=> other.d
+      when Float::INFINITY; return d <=> 1
+      when -Float::INFINITY; return d <=> -1
       when Numeric; return d
       else
         begin

diff --git a/test/date/test_date.rb b/test/date/test_date.rb
@@ -163,4 +163,16 @@ def test_submillisecond_comparison
     assert_equal(1, d2 <=> d1)
   end
 
+  def test_infinity_comparison
+    assert_equal(0, Float::INFINITY <=> Date::Infinity.new)
+    assert_equal(0, Date::Infinity.new <=> Float::INFINITY)
+    assert_equal(0, -Float::INFINITY <=> -Date::Infinity.new)
+    assert_equal(0, -Date::Infinity.new <=> -Float::INFINITY)
+
+    assert_equal(1, Float::INFINITY <=> -Date::Infinity.new)
+    assert_equal(1, Date::Infinity.new <=> -Float::INFINITY)
+
+    assert_equal(-1, -Float::INFINITY <=> Date::Infinity.new)
+    assert_equal(-1, -Date::Infinity.new <=> Float::INFINITY)
+  end
 end