Exemplo n.º 1
0
  def testGithubIssue18(self):
    d = [0, 1, 2, 3, 4]
    a = [0, 0, 1, 1, 1]
    tpl = Quantize.FindVarMultQuantBounds(d, 1, a, 2)

    d2 = [(x, ) for x in d]
    self.assertRaises(ValueError, lambda: Quantize.FindVarMultQuantBounds(d2, 1, a, 2))
    self.assertRaises(ValueError, lambda: Quantize._FindStartPoints(d2, a, len(d2)))
Exemplo n.º 2
0
  def test9NewSplits(self):
    """ 
    """
    d = [(0, 0),
         (1, 1),
         (2, 0), ]
    varValues, resCodes = zip(*d)
    nPossibleRes = 2
    res = Quantize._NewPyFindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [1, 2], str(res))
    res = Quantize._FindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [1, 2], str(res))

    d = [(0, 1),
         (1, 0),
         (2, 1), ]
    varValues, resCodes = zip(*d)
    nPossibleRes = 2
    res = Quantize._NewPyFindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [1, 2], str(res))
    res = Quantize._FindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [1, 2], str(res))

    d = [(0, 0),
         (0, 0),
         (1, 1),
         (1, 1),
         (2, 0),
         (2, 1), ]
    varValues, resCodes = zip(*d)
    nPossibleRes = 2
    res = Quantize._NewPyFindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [2, 4], str(res))
    res = Quantize._FindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [2, 4], str(res))

    d = [(0, 0),
         (0, 1),
         (1, 1),
         (1, 1),
         (2, 0),
         (2, 1), ]
    varValues, resCodes = zip(*d)
    nPossibleRes = 2
    res = Quantize._NewPyFindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [2, 4], str(res))
    res = Quantize._FindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [2, 4], str(res))

    d = [(0, 0),
         (0, 0),
         (1, 0),
         (1, 1),
         (2, 0),
         (2, 1), ]
    varValues, resCodes = zip(*d)
    nPossibleRes = 2
    res = Quantize._NewPyFindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [2, 4], str(res))
    res = Quantize._FindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [2, 4], str(res))

    d = [(0, 0),
         (0, 0),
         (1, 0),
         (1, 0),
         (2, 1),
         (2, 1), ]
    varValues, resCodes = zip(*d)
    nPossibleRes = 2
    res = Quantize._NewPyFindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [4], str(res))
    res = Quantize._FindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [4], str(res))

    d = [(0, 0),
         (0, 0),
         (1, 1),
         (1, 1),
         (2, 1),
         (2, 1), ]
    varValues, resCodes = zip(*d)
    nPossibleRes = 2
    res = Quantize._NewPyFindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [2], str(res))
    res = Quantize._FindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [2], str(res))

    d = [(0, 0),
         (0, 0),
         (1, 0),
         (1, 0),
         (2, 0),
         (2, 0), ]
    varValues, resCodes = zip(*d)
    nPossibleRes = 2
    res = Quantize._NewPyFindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [], str(res))
    res = Quantize._FindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [], str(res))

    d = [(0, 0),
         (0, 1),
         (1, 0),
         (1, 1),
         (2, 0),
         (2, 0), ]
    varValues, resCodes = zip(*d)
    nPossibleRes = 2
    res = Quantize._NewPyFindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [2, 4], str(res))
    res = Quantize._FindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [2, 4], str(res))

    d = [(1, 0),
         (2, 1),
         (2, 1),
         (3, 1),
         (3, 1),
         (3, 1),
         (4, 0),
         (4, 1),
         (4, 1), ]
    varValues, resCodes = zip(*d)
    nPossibleRes = 2
    res = Quantize._NewPyFindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [1, 6], str(res))
    res = Quantize._FindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [1, 6], str(res))

    d = [(1, 1.65175902843, 0), (2, 1.89935600758, 0), (3, 1.89935600758, 1), (4, 1.89935600758, 1),
         (5, 2.7561609745, 1), (6, 2.7561609745, 1), (7, 2.7561609745, 1), (8, 2.7561609745, 1),
         (9, 3.53454303741, 1), (10, 3.53454303741, 1), (11, 3.53454303741, 1),
         (12, 3.53454303741, 1), (13, 3.53454303741, 1)]

    _, varValues, resCodes = zip(*d)
    nPossibleRes = 2
    res = Quantize._NewPyFindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [1, 4], str(res))
    res = Quantize._FindStartPoints(varValues, resCodes, len(d))
    self.assertTrue(res == [1, 4], str(res))