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)
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)