Skip to content

Commit

Permalink
widen Graph.__contains___ triple type hint
Browse files Browse the repository at this point in the history
Paths are valid as the second argument of the triple here, as ultimately
the triples method is called, which accepts either a predicate or a
Path.
  • Loading branch information
EFord36 committed Mar 28, 2023
1 parent 40be70e commit c7bbb64
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions rdflib/graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -675,7 +675,7 @@ def __iter__(self) -> Generator["_TripleType", None, None]:
"""Iterates over all triples in the store"""
return self.triples((None, None, None))

def __contains__(self, triple: _TriplePatternType) -> bool:
def __contains__(self, triple: _TripleSelectorType) -> bool:
"""Support for 'triple in graph' syntax"""
for triple in self.triples(triple):
return True
Expand Down Expand Up @@ -1977,7 +1977,7 @@ def _spoc(
c = self._graph(c)
return s, p, o, c

def __contains__(self, triple_or_quad: _TripleOrQuadPatternType) -> bool:
def __contains__(self, triple_or_quad: _TripleOrQuadSelectorType) -> bool:
"""Support for 'triple/quad in graph' syntax"""
s, p, o, c = self._spoc(triple_or_quad)
for t in self.triples((s, p, o), context=c):
Expand Down Expand Up @@ -2751,7 +2751,7 @@ def triples(
for s1, p1, o1 in graph.triples((s, p, o)):
yield s1, p1, o1

def __contains__(self, triple_or_quad: _TripleOrQuadPatternType) -> bool:
def __contains__(self, triple_or_quad: _TripleOrQuadSelectorType) -> bool:
context = None
if len(triple_or_quad) == 4:
# type error: Tuple index out of range
Expand Down

0 comments on commit c7bbb64

Please sign in to comment.