def test_split_bin(self): dg = DataGenerator() _bin = [21, 41, (0, 0)] self.assertEqual(dg._split_bin(_bin, 0, 14), [[14, 41, (0, 0)], [21 - 14, 41, (14, 0)]]) self.assertEqual(dg._split_bin(_bin, 1, 14), [[21, 14, (0, 0)], [21, 41 - 14, (0, 14)]])