def test_adapt_internal(self):
b1 = binnings.FixedWidthBinning(bin_width=10,
bin_count=3,
min=0,
adaptive=True)
b2 = binnings.FixedWidthBinning(bin_width=10,
bin_count=1,
min=10,
adaptive=True)
m1, m2 = b1.adapt(b2)
assert m1 is None
assert tuple(m2) == ((0, 1), )
def test_adapt_intersection2(self):
b = binnings.FixedWidthBinning(bin_width=10,
bin_count=3,
min=0,
adaptive=True)
b6 = binnings.FixedWidthBinning(bin_width=10,
bin_count=3,
min=10,
adaptive=True)
m1, m2 = b6.adapt(b)
assert tuple(m1) == ((0, 1), (1, 2), (2, 3))
assert tuple(m2) == ((0, 0), (1, 1), (2, 2))
assert b6.bin_count == 4
def test_adapt_right(self):
b = binnings.FixedWidthBinning(bin_width=10,
bin_count=3,
min=0,
adaptive=True)
b4 = binnings.FixedWidthBinning(bin_width=10,
bin_count=2,
min=-30,
adaptive=True)
m1, m2 = b4.adapt(b)
assert tuple(m1) == ((0, 0), (1, 1))
assert tuple(m2) == ((0, 3), (1, 4), (2, 5))
assert b4.bin_count == 6
def test_adapt_left(self):
b = binnings.FixedWidthBinning(bin_width=10,
bin_count=3,
min=0,
adaptive=True)
b3 = binnings.FixedWidthBinning(bin_width=10,
bin_count=2,
min=50,
adaptive=True)
m1, m2 = b3.adapt(b)
assert tuple(m1) == ((0, 5), (1, 6))
assert tuple(m2) == ((0, 0), (1, 1), (2, 2))
assert b3.bin_count == 7
def test_adapt_extension(self):
b = binnings.FixedWidthBinning(bin_width=10,
bin_count=3,
min=0,
adaptive=True)
b2 = binnings.FixedWidthBinning(bin_width=10,
bin_count=2,
min=0,
adaptive=True)
m1, m2 = b2.adapt(b)
assert tuple(m1) == ((0, 0), (1, 1))
assert m2 is None
assert np.array_equal(b2.numpy_bins, [0, 10, 20, 30])
assert b2.bin_count == 3
def test_adapt_wrong(self):
b1 = binnings.FixedWidthBinning(bin_width=10,
bin_count=2,
min=0,
adaptive=True)
b2 = binnings.FixedWidthBinning(bin_width=10,
bin_count=2,
min=1,
adaptive=True)
with self.assertRaises(RuntimeError):
b1.adapt(b2)
with self.assertRaises(RuntimeError):
b2.adapt(b1)
b3 = binnings.FixedWidthBinning(bin_width=5,
bin_count=6,
min=0,
adaptive=True)
with self.assertRaises(RuntimeError):
b1.adapt(b3)
with self.assertRaises(RuntimeError):
b3.adapt(b1)