예제 #1
0
    def test_filtermarkers_three_list_one_missing_markerlist(self):
        mlist1 = MarkersList()
        mlist1.add_marker(1, 2, 3, 4)
        mlist1.add_marker(1, 2, 3, 5)
        mlist1.add_marker(2, 2, 3, 6)
        mlist1.add_marker(4, 2, 3, 6)
        mlist1.add_marker(4, 2, 3, 7)

        thelist = []
        thelist.append(mlist1)

        mlist2 = MarkersList()
        mlist2.add_marker(1, 2, 3, 4)
        mlist2.add_marker(1, 2, 3, 5)
        mlist2.add_marker(3, 2, 3, 6)
        mlist2.add_marker(4, 2, 3, 6)
        mlist2.add_marker(4, 2, 3, 7)
        thelist.append(mlist2)

        mlist3 = MarkersList()
        mlist3.add_marker(1, 2, 3, 4)
        mlist3.add_marker(1, 2, 3, 5)
        mlist3.add_marker(3, 2, 3, 6)
        mlist3.add_marker(4, 2, 3, 6)
        mlist3.add_marker(5, 1, 1, 1)
        thelist.append(mlist3)

        filt = CommonByIndexMarkersListFilter(thelist)

        com, uni = filt.filterMarkers(mlist1)

        self.assertTrue(len(uni.get_markers()) == 1)
        self.assertTrue(len(com.get_markers()) == 4)

        self.assertEqual(uni.get_markers()[0].get_index(), 2)
        self.assertEqual(uni.get_markers()[0].get_z(), 6)

        com, uni = filt.filterMarkers(mlist2)

        self.assertTrue(len(uni.get_markers()) == 1)
        self.assertTrue(len(com.get_markers()) == 4)

        self.assertEqual(uni.get_markers()[0].get_index(), 3)
        self.assertEqual(uni.get_markers()[0].get_y(), 3)

        com, uni = filt.filterMarkers(mlist3)

        self.assertTrue(len(uni.get_markers()) == 2)
        self.assertTrue(len(com.get_markers()) == 3)

        self.assertEqual(uni.get_markers()[0].get_index(), 3)
        self.assertEqual(uni.get_markers()[0].get_y(), 3)
        self.assertEqual(uni.get_markers()[1].get_index(), 5)
        self.assertEqual(uni.get_markers()[1].get_y(), 1)
예제 #2
0
    def test_filtermarkers_one_markerlist(self):
        mlist = MarkersList()
        mlist.add_marker(1, 2, 3, 4)
        mlist.add_marker(1, 2, 3, 5)
        mlist.add_marker(1, 2, 3, 6)
        thelist = []
        thelist.append(mlist)
        filt = CommonByIndexMarkersListFilter(thelist)

        com, uni = filt.filterMarkers(mlist)

        self.assertTrue(len(uni.get_markers()) == 0)
        self.assertTrue(len(com.get_markers()) == 3)
        m = com.get_markers()[0]
        self.assertEqual(m.get_index(), 1)
        self.assertEqual(m.get_x(), 2)
        self.assertEqual(m.get_y(), 3)
        self.assertEqual(m.get_z(), 4)