def test_floor(self): from numpypy import array, floor reference = [-2.0, -1.0, 0.0, 1.0, 1.0] a = array([-1.4, -1.0, 0.0, 1.0, 1.4]) b = floor(a) for i in range(5): assert b[i] == reference[i]
def test_floorceiltrunc(self): from numpypy import array, floor, ceil, trunc import math ninf, inf = float("-inf"), float("inf") a = array([ninf, -1.4, -1.5, -1.0, 0.0, 1.0, 1.4, 0.5, inf]) assert ([ninf, -2.0, -2.0, -1.0, 0.0, 1.0, 1.0, 0.0, inf] == floor(a)).all() assert ([ninf, -1.0, -1.0, -1.0, 0.0, 1.0, 2.0, 1.0, inf] == ceil(a)).all() assert ([ninf, -1.0, -1.0, -1.0, 0.0, 1.0, 1.0, 0.0, inf] == trunc(a)).all() assert all([math.isnan(f(float("nan"))) for f in floor, ceil, trunc])