def test_next_None(self):
binning = Echo(nextFunc=None)
self.assertIsNone(binning.next(-5))
self.assertIsNone(binning.next(0))
self.assertIsNone(binning.next(1))
self.assertIsNone(binning.next(3))
self.assertIsNone(binning.next(10))
def test_next_plus2(self):
binning = Echo(nextFunc=plus2)
self.assertEqual(-3, binning.next(-5))
self.assertEqual(2, binning.next(0))
self.assertEqual(3, binning.next(1))
self.assertEqual(5, binning.next(3))
self.assertEqual(12, binning.next(10))
def test_next_lambda(self):
binning = Echo(nextFunc=lambda x: x + 0.1)
self.assertEqual(-4.9, binning.next(-5))
self.assertEqual(0.1, binning.next(0))
self.assertEqual(1.1, binning.next(1))
self.assertEqual(3.1, binning.next(3))
self.assertEqual(10.1, binning.next(10))
def test_next_default(self):
binning = Echo()
self.assertEqual(-4, binning.next(-5))
self.assertEqual(1, binning.next(0))
self.assertEqual(2, binning.next(1))
self.assertEqual(4, binning.next(3))
self.assertEqual(11, binning.next(10))