def test_extremes(): tensor = DenseTensor(zeros(100,)) for i in range(100): tensor[i] = i*10 extremes = tensor.extremes() bottom, top = extremes[:10], extremes[10:] eq_(bottom, [((i,), i*10.) for i in range(10)]) eq_(top, [((i,), i*10.) for i in range(90,100)])
def test_extremes(): tensor = DenseTensor(zeros(100, )) for i in range(100): tensor[i] = i * 10 extremes = tensor.extremes() bottom, top = extremes[:10], extremes[10:] eq_(bottom, [((i, ), i * 10.) for i in range(10)]) eq_(top, [((i, ), i * 10.) for i in range(90, 100)])
def test_extreme_items(): tensor = DenseTensor(zeros(100,)) for i in range(100): tensor[i] = i*10 extremes = tensor.extremes() top, bottom = tensor.extreme_items(10, 10) eq_(bottom, [((i,), i*10.) for i in range(10)]) eq_(top, [((i,), i*10.) for i in range(90,100)])
def test_extreme_items(): tensor = DenseTensor(zeros(100, )) for i in range(100): tensor[i] = i * 10 extremes = tensor.extremes() top, bottom = tensor.extreme_items(10, 10) eq_(bottom, [((i, ), i * 10.) for i in range(10)]) eq_(top, [((i, ), i * 10.) for i in range(90, 100)])