use malachite_base::num::arithmetic::traits::{AbsSquared, Conjugate};
use malachite_base::num::basic::traits::{Infinity, NaN, NegativeInfinity};
use malachite_base::num::comparison::traits::EqAbs;
use malachite_base::num::conversion::traits::ExactFrom;
use malachite_float::Float;
use malachite_float::test_util::generators::{
float_gaussian_rational_pair_gen, float_rational_pair_gen,
};
use malachite_q::Rational;
use malachite_q::gaussian_rational::GaussianRational;
use malachite_q::test_util::generators::gaussian_rational_gen;
use std::str::FromStr;
#[test]
fn test_eq_abs_gaussian_rational() {
let test = |s, t, out| {
let x = Float::from_str(s).unwrap();
let y = GaussianRational::from_str(t).unwrap();
assert_eq!(x.eq_abs(&y), out);
assert_eq!(y.eq_abs(&x), out);
};
test("0.0", "0", true);
test("-0.0", "0", true);
test("1.0", "i", true);
test("-0.5", "i/2", true);
test("1.0", "3/5+4i/5", true);
test("-1.0", "3/5+4i/5", true);
test("2.5", "3/2+2i", true);
test("-2.5", "3/2+2i", true);
test("2.0", "3/2+2i", false);
test("0.5", "3/10+2i/5", true);
test("5.0", "3+4i", true);
test("3.0", "2+2i", false);
test("NaN", "0", false);
test("Infinity", "3/2+2i", false);
test("-Infinity", "0", false);
}
#[test]
fn eq_abs_gaussian_rational_properties() {
float_gaussian_rational_pair_gen().test_properties(|(x, y)| {
let eq = x.eq_abs(&y);
assert_eq!(y.eq_abs(&x), eq);
assert_eq!((-&x).eq_abs(&y), eq);
assert_eq!(x.eq_abs(&-&y), eq);
assert_eq!(x.eq_abs(&(&y).conjugate()), eq);
if x.is_finite() {
assert_eq!(
Rational::exact_from(&x).abs_squared() == (&y).abs_squared(),
eq
);
} else {
assert!(!eq);
}
});
gaussian_rational_gen().test_properties(|x| {
assert_eq!(x.eq_abs(&Float::NAN), false);
assert_eq!(Float::NAN.eq_abs(&x), false);
assert_eq!(x.eq_abs(&Float::INFINITY), false);
assert_eq!(x.eq_abs(&Float::NEGATIVE_INFINITY), false);
});
float_rational_pair_gen().test_properties(|(x, y)| {
assert_eq!(x.eq_abs(&GaussianRational::from(y.clone())), x.eq_abs(&y));
assert_eq!(GaussianRational::from(y.clone()).eq_abs(&x), y.eq_abs(&x));
});
}