diff options
author | Kenta Murata <3959+mrkn@users.noreply.github.com> | 2024-02-27 09:37:03 +0900 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-02-27 09:37:03 +0900 |
commit | 54a5b829442eb7ed1f4a2794ebb26696cf784e64 (patch) | |
tree | 1bdb2b9212fc19faf9e7babd3e6324d6fb10aaee /complex.c | |
parent | 89f0c0ceb5e5e20c56e63498521966c499b361a5 (diff) |
Handle zero-like imaginary part as zero in to_r (#9581)
Fixes [Bug #5179]
Diffstat (limited to 'complex.c')
-rw-r--r-- | complex.c | 16 |
1 files changed, 12 insertions, 4 deletions
@@ -1841,9 +1841,11 @@ nucomp_to_f(VALUE self) * * Complex.rect(1, 0).to_r # => (1/1) * Complex.rect(1, Rational(0, 1)).to_r # => (1/1) + * Complex.rect(1, 0.0).to_r # => (1/1) * * Raises RangeError if <tt>self.imag</tt> is not exactly zero - * (either <tt>Integer(0)</tt> or <tt>Rational(0, _n_)</tt>). + * (either <tt>Integer(0)</tt> or <tt>Rational(0, _n_)</tt>) + * and <tt>self.imag.to_r</tt> is not exactly zero. * * Related: Complex#rationalize. */ @@ -1852,9 +1854,15 @@ nucomp_to_r(VALUE self) { get_dat1(self); - if (!k_exact_zero_p(dat->imag)) { - rb_raise(rb_eRangeError, "can't convert %"PRIsVALUE" into Rational", - self); + if (RB_FLOAT_TYPE_P(dat->imag) && FLOAT_ZERO_P(dat->imag)) { + /* Do nothing here */ + } + else if (!k_exact_zero_p(dat->imag)) { + VALUE imag = rb_check_convert_type_with_id(dat->imag, T_RATIONAL, "Rational", idTo_r); + if (NIL_P(imag) || !k_exact_zero_p(imag)) { + rb_raise(rb_eRangeError, "can't convert %"PRIsVALUE" into Rational", + self); + } } return f_to_r(dat->real); } |