// Copyright © 2024 Mikhail Hogrefe
//
// This file is part of Malachite.
//
// Malachite is free software: you can redistribute it and/or modify it under the terms of the GNU
// Lesser General Public License (LGPL) as published by the Free Software Foundation; either version
// 3 of the License, or (at your option) any later version. See <https://www.gnu.org/licenses/>.

use malachite_base::num::basic::integers::PrimitiveInt;
use malachite_base::num::basic::signeds::PrimitiveSigned;
use malachite_base::num::basic::unsigneds::PrimitiveUnsigned;
use malachite_base::num::conversion::traits::ExactFrom;
use malachite_base::test_util::generators::{signed_gen, signed_gen_var_2, unsigned_gen};
use malachite_nz::natural::Natural;
use malachite_nz::platform::{Limb, SignedLimb};
use malachite_q::Rational;
use rug;

#[test]
fn test_from_u32() {
    let test = |u: u32, out| {
        let x = Rational::from(u);
        assert_eq!(x.to_string(), out);
        assert!(x.is_valid());
        assert_eq!(rug::Rational::from(u).to_string(), out);
        #[cfg(feature = "32_bit_limbs")]
        {
            let x_alt = Rational::const_from_unsigned(u);
            assert!(x_alt.is_valid());
            assert_eq!(x_alt.to_string(), out);
        }
    };
    test(0, "0");
    test(123, "123");
    test(u32::MAX, "4294967295");
}

#[test]
fn test_from_u64() {
    let test = |u: u64, out| {
        let x = Rational::from(u);
        assert_eq!(x.to_string(), out);
        assert!(x.is_valid());
        assert_eq!(rug::Rational::from(u).to_string(), out);
        #[cfg(not(feature = "32_bit_limbs"))]
        {
            let x_alt = Rational::const_from_unsigned(u);
            assert!(x_alt.is_valid());
            assert_eq!(x_alt.to_string(), out);
        }
    };
    test(0, "0");
    test(123, "123");
    test(u64::MAX, "18446744073709551615");
}

#[test]
fn test_from_i32() {
    let test = |i: i32, out| {
        let x = Rational::from(i);
        assert_eq!(x.to_string(), out);
        assert!(x.is_valid());
        assert_eq!(rug::Rational::from(i).to_string(), out);
        #[cfg(feature = "32_bit_limbs")]
        {
            let x_alt = Rational::const_from_signed(i);
            assert!(x_alt.is_valid());
            assert_eq!(x_alt.to_string(), out);
        }
    };
    test(0, "0");
    test(123, "123");
    test(-123, "-123");
    test(i32::MIN, "-2147483648");
    test(i32::MAX, "2147483647");
}

#[test]
fn test_from_i64() {
    let test = |i: i64, out| {
        let x = Rational::from(i);
        assert_eq!(x.to_string(), out);
        assert!(x.is_valid());
        assert_eq!(rug::Rational::from(i).to_string(), out);
        #[cfg(not(feature = "32_bit_limbs"))]
        {
            let x_alt = Rational::const_from_signed(i);
            assert!(x_alt.is_valid());
            assert_eq!(x_alt.to_string(), out);
        }
    };
    test(0, "0");
    test(123, "123");
    test(-123, "-123");
    test(i64::MIN, "-9223372036854775808");
    test(i64::MAX, "9223372036854775807");
}

fn from_unsigned_properties_helper<T: for<'a> TryFrom<&'a Rational> + PrimitiveUnsigned>()
where
    Rational: From<T>,
    Natural: From<T>,
    u128: TryFrom<T>,
    rug::Integer: From<T>,
    Limb: ExactFrom<T>,
{
    unsigned_gen::<T>().test_properties(|u| {
        let n = Rational::from(u);
        assert!(n.is_valid());
        assert_eq!(T::exact_from(&n), u);
        let alt_n: Rational = From::from(Natural::from(u));
        assert_eq!(alt_n, n);
        let alt_n: Rational = From::from(u128::exact_from(u));
        assert_eq!(alt_n, n);
        let alt_n: Rational = From::from(&rug::Rational::from(u));
        assert_eq!(alt_n, n);
        if T::WIDTH == Limb::WIDTH {
            let n_alt = Rational::const_from_unsigned(Limb::exact_from(u));
            assert!(n_alt.is_valid());
            assert_eq!(n_alt, n);
        }
    });
}

fn from_signed_properties_helper<T: for<'a> TryFrom<&'a Rational> + PrimitiveSigned>()
where
    Rational: From<T>,
    Natural: TryFrom<T>,
    i128: TryFrom<T>,
    rug::Integer: From<T>,
    SignedLimb: ExactFrom<T>,
{
    signed_gen::<T>().test_properties(|i| {
        let n = Rational::from(i);
        assert!(n.is_valid());
        assert_eq!(T::exact_from(&n), i);
        let alt_n: Rational = From::from(i128::exact_from(i));
        assert_eq!(alt_n, n);
        let alt_n: Rational = From::from(&rug::Rational::from(i));
        assert_eq!(alt_n, n);
        if T::WIDTH == Limb::WIDTH {
            let n_alt = Rational::const_from_signed(SignedLimb::exact_from(i));
            assert!(n_alt.is_valid());
            assert_eq!(n_alt, n);
        }
    });

    signed_gen_var_2::<T>().test_properties(|i| {
        let n: Rational = From::from(Natural::exact_from(i));
        assert_eq!(n, Rational::from(i));
    });
}

#[test]
fn from_primitive_int_properties() {
    apply_fn_to_unsigneds!(from_unsigned_properties_helper);
    apply_fn_to_signeds!(from_signed_properties_helper);
}