def test_cut_none(self):
a = HistogramAxis(10, [0, 10])
assertaaeq = assert_array_almost_equal
assertaaeq(a.cut(None, 3)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(7, None)[0].edges, np.linspace(7, 10, 4))
with self.assertRaises(TypeError):
a.cut(0, 3, 0.)
def test_cut(self):
a = HistogramAxis(10,[0,10])
assert_array_almost_equal(a.cut(-1,3)[0].edges, np.linspace(0,3,4))
assert_array_almost_equal(a.cut(7,11)[0].edges, np.linspace(7,10,4))
assert_array_almost_equal(a.cut(-0.1,3)[0].edges, np.linspace(0,3,4))
assert_array_almost_equal(a.cut( 0.0,3)[0].edges, np.linspace(0,3,4))
assert_array_almost_equal(a.cut( 0.1,3)[0].edges, np.linspace(0,3,4))
assert_array_almost_equal(a.cut(2.9,6.9)[0].edges, np.linspace(3,7,5))
assert_array_almost_equal(a.cut(3.0,7.0)[0].edges, np.linspace(3,7,5))
assert_array_almost_equal(a.cut(3.1,7.1)[0].edges, np.linspace(3,7,5))
assert_array_almost_equal(a.cut(3.1,7.49999)[0].edges, np.linspace(3,7,5))
assert_array_almost_equal(a.cut(3.1,7.5)[0].edges, np.linspace(3,8,6))
def test_cut_clip(self):
a = HistogramAxis(10, [0, 10])
assertaaeq = assert_array_almost_equal
snap = 'clip'
assertaaeq(a.cut(-1, 3, snap)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(7, 11, snap)[0].edges, np.linspace(7, 10, 4))
assertaaeq(a.cut(-0.1, 3, snap)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(0.0, 3, snap)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(0.1, 3, snap)[0].edges, [0.1, 1, 2, 3])
assertaaeq(a.cut(2.9, 6.9, snap)[0].edges, [2.9, 3, 4, 5, 6, 6.9])
assertaaeq(a.cut(3.0, 7.0, snap)[0].edges, np.linspace(3, 7, 5))
assertaaeq(a.cut(3.1, 7.1, snap)[0].edges, [3.1, 4, 5, 6, 7, 7.1])
assertaaeq(a.cut(3.1, 5.49999, snap)[0].edges, [3.1, 4, 5, 5.49999])
assertaaeq(a.cut(3.1, 5.5, snap)[0].edges, [3.1, 4, 5, 5.5])
def test_cut_high(self):
a = HistogramAxis(10, [0, 10])
assertaaeq = assert_array_almost_equal
snap = 'high'
assertaaeq(a.cut(-1, 3, snap)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(7, 11, snap)[0].edges, np.linspace(7, 10, 4))
assertaaeq(a.cut(-0.1, 3, snap)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(0.0, 3, snap)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(0.1, 3, snap)[0].edges, np.linspace(1, 3, 3))
assertaaeq(a.cut(2.9, 6.9, snap)[0].edges, np.linspace(3, 7, 5))
assertaaeq(a.cut(3.0, 7.0, snap)[0].edges, np.linspace(3, 7, 5))
assertaaeq(a.cut(3.1, 7.1, snap)[0].edges, np.linspace(4, 8, 5))
assertaaeq(a.cut(3.1, 7.49999, snap)[0].edges, np.linspace(4, 8, 5))
assertaaeq(a.cut(3.1, 7.5, snap)[0].edges, np.linspace(4, 8, 5))
def test_cut_nearest(self):
a = HistogramAxis(10, [0, 10])
assertaaeq = assert_array_almost_equal
assertaaeq(a.cut(-1, 3)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(7, 11)[0].edges, np.linspace(7, 10, 4))
assertaaeq(a.cut(-0.1, 3)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(0.0, 3)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(0.1, 3)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(2.9, 6.9)[0].edges, np.linspace(3, 7, 5))
assertaaeq(a.cut(3.0, 7.0)[0].edges, np.linspace(3, 7, 5))
assertaaeq(a.cut(3.1, 7.1)[0].edges, np.linspace(3, 7, 5))
assertaaeq(a.cut(3.1, 7.49999)[0].edges, np.linspace(3, 7, 5))
assertaaeq(a.cut(3.1, 7.5)[0].edges, np.linspace(3, 8, 6))
if __debug__:
with self.assertRaises(ValueError):
a.cut(0, 1, 'x')
# 'nearest','expand','low','high','clip'
snap = 'nearest'
assertaaeq(a.cut(-1, 3, snap)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(7, 11, snap)[0].edges, np.linspace(7, 10, 4))
assertaaeq(a.cut(-0.1, 3, snap)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(0.0, 3, snap)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(0.1, 3, snap)[0].edges, np.linspace(0, 3, 4))
assertaaeq(a.cut(2.9, 6.9, snap)[0].edges, np.linspace(3, 7, 5))
assertaaeq(a.cut(3.0, 7.0, snap)[0].edges, np.linspace(3, 7, 5))
assertaaeq(a.cut(3.1, 7.1, snap)[0].edges, np.linspace(3, 7, 5))
assertaaeq(a.cut(3.1, 7.49999, snap)[0].edges, np.linspace(3, 7, 5))
assertaaeq(a.cut(3.1, 7.5, snap)[0].edges, np.linspace(3, 8, 6))
