1.16 - Implementação e asserts

vbuxbaum · Apr 2, 2021 · 4ec2034 · 4ec2034
1 parent fb0521a
commit 4ec2034
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/1_introducao/16_meu_item.py b/1_introducao/16_meu_item.py
@@ -0,0 +1,23 @@
+class Item1(object):
+    def __gt__(self, outro_item):
+        raise NotImplementedError()
+
+
+class MeuItem(Item1):
+    def __init__(self, chave):
+        self.chave = chave
+
+    def __gt__(self, outro_item):
+        return self.chave > outro_item.chave
+
+
+if __name__ == "__main__":
+    item_teste_1 = MeuItem(321)
+    item_teste_2 = MeuItem(123)
+
+    assert item_teste_1 > item_teste_2
+
+    item_teste_3 = MeuItem('321')
+    item_teste_4 = MeuItem('123')
+
+    assert item_teste_3 > item_teste_4