From 7cae00f98727dfb9a3d7f1f53e85ab8f7ee602b0 Mon Sep 17 00:00:00 2001
From: ksss <co000ri@gmail.com>
Date: Wed, 27 Mar 2024 23:19:55 +0900
Subject: [PATCH] `.int_from_prime_division` could return Rational.

---
 sig/prime.rbs    | 4 ++--
 test/test_rbs.rb | 4 ++++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/sig/prime.rbs b/sig/prime.rbs
index a4038ad..ee97938 100644
--- a/sig/prime.rbs
+++ b/sig/prime.rbs
@@ -134,7 +134,7 @@ class Prime
   #     Prime.int_from_prime_division([[3, 2], [5, 1]])  #=> 45
   #     3**2 * 5                                         #=> 45
   #
-  def self.int_from_prime_division: (Array[[ Integer, Integer ]]) -> Integer
+  def self.int_from_prime_division: (Array[[ Integer, Integer ]]) -> (Integer | Rational)
 
   # <!--
   #   rdoc-file=lib/prime.rb
@@ -160,7 +160,7 @@ class Prime
   #     Prime.int_from_prime_division([[3, 2], [5, 1]])  #=> 45
   #     3**2 * 5                                         #=> 45
   #
-  def int_from_prime_division: (Array[[ Integer, Integer ]]) -> Integer
+  def int_from_prime_division: (Array[[ Integer, Integer ]]) -> (Integer | Rational)
 
   # <!--
   #   rdoc-file=lib/prime.rb
diff --git a/test/test_rbs.rb b/test/test_rbs.rb
index 8fd7c23..e123105 100644
--- a/test/test_rbs.rb
+++ b/test/test_rbs.rb
@@ -39,6 +39,8 @@ def test_each
     def test_int_from_prime_division
       assert_send_type "(::Array[[ ::Integer, ::Integer ]]) -> ::Integer",
                        Prime, :int_from_prime_division, [[3, 1], [19, 1]]
+      assert_send_type "(::Array[[ ::Integer, ::Integer ]]) -> ::Rational",
+                       Prime, :int_from_prime_division, [[-4, -2]]
     end
 
     def test_prime?
@@ -85,6 +87,8 @@ def test_each
     def test_int_from_prime_division
       assert_send_type "(::Array[[ ::Integer, ::Integer ]]) -> ::Integer",
                        Prime.instance, :int_from_prime_division, [[3, 1], [19, 1]]
+      assert_send_type "(::Array[[ ::Integer, ::Integer ]]) -> ::Rational",
+                       Prime.instance, :int_from_prime_division, [[-4, -2]]
     end
 
     def test_prime?