힙 내부의 객체인지 확인

Question

객체가 힙 내부에 있는지 확인하려고합니다.

이

import heapq 

class Heap(object): 
    def __init__(self): 
     self.heap = [] 

    def push(self, priority, cell): 
     heapq.heappush(self.heap, (priority, cell)) 

    def pop(self): 
     cell = heapq.heappop(self.heap)[1] 
     return cell 

    def contains(self, cell): 
     if(cell in self.heap): 
      return True 
     return False 

    def isEmpty(self): 
     return len(self.heap) == 0 

이 셀 클래스 :

class Cell(object): 
    def __init__(self, x, y): 
     self.X = x 
     self.Y = y 

    def __eq__(self, other): 
     return int(self.X) == int(other.X) and int(self.Y) == int(other.Y) 

내가 Heap 클래스를 사용 나는이 같은 봐 Heap과 Cell 클래스가

AttributeError: 'tuple' object has no attribute 'X' 

: 그러나, 나는이 오류가 계속 이렇게 : contains 메서드를 사용할 때 오류가 발생합니다.

from Heap import Heap 
from Cell import Cell 

class Test(object): 
    def __init__(self): 
     self.myHeap = Heap() 
     cell = Cell(2, 3) 

     self.myHeap.push(1, cell) 

     if self.myHeap.contains(cell) == False: 
      print("not in heap") 

test = Test() 

내가 뭘 잘못하고 있니? 어떤 도움을 주시면 감사하겠습니다.

Answer 1

문제는 contains 방법입니다.

def contains(self, cell): 
    if(cell in self.heap): 
     return True 
    return False 

self.head 형 (priority, Cell) 튜플들의 목록이다. 그리고 실제로 Cells을이 목록의 요소 (튜플)와 비교하므로 Cell.__eq__() 메서드가 호출되고 예외가 발생합니다.