티스토리 뷰


인스턴스에 대한 멤버십 테스트가 필요할때

미리 정의된 스페셜 함수인 __contains__ 를 이용해서 구현할 수 있다.


함수 인자로 아이템을 받고 검사하고자 하는 컨테이너의 멤버인지 확인해보자

그 논리 값을 리턴하여 보자.



class Coo:
def __init__(self, info=None):
self.info = info if info else list()

def add(self, item):
self.info.append(item)

def remove(self, index):
self.info.pop(index)

def __contains__(self, item):
return item in self.info


3은 self.info 에 없으므로 출력이 되지 않고, 1, 2는 포함되므로 include 를 출력한다.

c = Coo()
c.add(1)
c.add(2)
if 3 in c:
print("include")









댓글