def sut():
return Entity(entity=Widget)
def test_relationships(sut: Entity):
relationships = sut.get_relationship_fields()
assert len(relationships) == 3
assert relationships[0].name == 'addresses'
assert relationships[1].name == 'category'
assert relationships[2].name == 'part'
def test_equality(sut: Entity):
assert sut == Widget
assert sut == sut
assert sut == Entity(Widget)
assert sut != Address
assert sut != Entity(Address)
def category():
return Entity(Category)
def part():
return Entity(Part)
def address():
return Entity(Address)
def widget():
return Entity(Widget)
def sut():
return Entity(Widget)