diff options
author | Stephen Ierodiaconou <stevegeek@gmail.com> | 2022-10-23 05:59:06 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-10-23 12:59:06 +0900 |
commit | 54cad3123a07583c90e85bcfc55ebd87124c1250 (patch) | |
tree | d6d1602e479d94813368f373ad7dc30d8f4cc6a9 /complex.c | |
parent | 0d9628e0decc6c2c7ca0ca4e7ecbf7e4d025959b (diff) | |
download | ruby-54cad3123a07583c90e85bcfc55ebd87124c1250.tar.gz |
[Bug #19004] `Complex.polar` handles complex singular `abs` argument
`Complex.polar` accepts Complex values as arguments for the polar form as long
as the value of the complex has no imaginary part (ie it is 'real'). In
`f_complex_polar` this is handled by extracting the real part of the arguments.
However in the case `polar` is called with only a single argument, the absolute
value (abs), then the Complex is created without applying a check on the type
of abs, meaning it is possible to create a Complex where the real part is itself
an instance of a Complex. This change removes the short circuit for the single
argument case meaning the real part extraction is performed correctly
(by f_complex_polar).
Also adds an example to `spec/ruby/core/complex/polar_spec.rb` to check that
the real part of a complex argument is correctly extracted and used in the
resulting Complex real and imaginary parts.
Diffstat (limited to 'complex.c')
-rw-r--r-- | complex.c | 13 |
1 files changed, 6 insertions, 7 deletions
@@ -704,14 +704,13 @@ nucomp_s_polar(int argc, VALUE *argv, VALUE klass) { VALUE abs, arg; - switch (rb_scan_args(argc, argv, "11", &abs, &arg)) { - case 1: - nucomp_real_check(abs); - return nucomp_s_new_internal(klass, abs, ZERO); - default: - nucomp_real_check(abs); + argc = rb_scan_args(argc, argv, "11", &abs, &arg); + nucomp_real_check(abs); + if (argc == 2) { nucomp_real_check(arg); - break; + } + else { + arg = ZERO; } return f_complex_polar(klass, abs, arg); } |