# coding: utf-8
# Licensed under a 3-clause BSD style license - see LICENSE.rst
"""
    Test utilities for `astropy.units`.
"""
import numpy as np
from numpy import finfo
from astropy.units.utils import sanitize_scale
from astropy.units.utils import quantity_asanyarray
from astropy.units.quantity import Quantity

_float_finfo = finfo(float)


def test_quantity_asanyarray():
    array_of_quantities = [Quantity(1), Quantity(2), Quantity(3)]
    quantity_array = quantity_asanyarray(array_of_quantities)
    assert isinstance(quantity_array, Quantity)

    array_of_integers = [1, 2, 3]
    np_array = quantity_asanyarray(array_of_integers)
    assert isinstance(np_array, np.ndarray)


def test_sanitize_scale():
    assert sanitize_scale(complex(2, _float_finfo.eps)) == 2
    assert sanitize_scale(complex(_float_finfo.eps, 2)) == 2j
