Exemplo n.º 1
0
    def test_reindexAfterMerge(self):
        MSelector = MergePairSelector()
        MSelector.MScores[0] = 5
        MSelector.MScores[3] = 5
        MSelector.MScores[4] = 5
        MSelector.PairMScores[(0, 1)] = 5
        MSelector.PairMScores[(3, 4)] = 5
        MSelector.PairMScores[(5, 6)] = 5

        MSelector.reindexAfterMerge(2, 3)

        assert MSelector.MScores[0] == 5
        assert MSelector.MScores[3] == 5
        assert 2 not in MSelector.MScores

        assert len(MSelector.PairMScores.keys()) == 2
        assert (0, 1) in MSelector.PairMScores
        assert (4, 5) in MSelector.PairMScores
        assert (2, 3) not in MSelector.PairMScores
  def test_reindexAfterMerge(self):
    MSelector = MergePairSelector()
    MSelector.MScores[0] = 5
    MSelector.MScores[3] = 5
    MSelector.MScores[4] = 5
    MSelector.PairMScores[(0,1)] = 5
    MSelector.PairMScores[(3,4)] = 5
    MSelector.PairMScores[(5,6)] = 5

    MSelector.reindexAfterMerge(2,3)

    assert MSelector.MScores[0] == 5
    assert MSelector.MScores[3] == 5
    assert 2 not in MSelector.MScores

    assert len(MSelector.PairMScores.keys()) == 2
    assert (0,1) in MSelector.PairMScores
    assert (4,5) in MSelector.PairMScores
    assert (2,3) not in MSelector.PairMScores