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))