Marginally increase test coverage

jarhill0 · Feb 17, 2018 · 8d1bd9f · 8d1bd9f
1 parent 7823c72
commit 8d1bd9f
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/tests/unit/models/main_models/test_photo_size.py b/tests/unit/models/main_models/test_photo_size.py
@@ -21,9 +21,38 @@ def test_eq():
     assert ps1 != same_size
     assert ps2 != same_size
 
+    different_size = PhotoSize(TG, dict(file_id=67890, width=640, height=1000))
+    assert different_size != ps1
+
 
 def test_gt():
     ps1 = PhotoSize(TG, DICT_1)
     bigger = PhotoSize(TG, dict(file_id=1, width=641, height=1000))
 
     assert bigger > ps1
+
+    try:
+        bigger > 3
+    except TypeError:
+        pass
+    else:
+        assert False, "These types are not comparable"
+
+    try:
+        bigger > '3'
+    except TypeError:
+        pass
+    else:
+        assert False, "These types are not comparable"
+
+    try:
+        bigger > Telegram
+    except TypeError:
+        pass
+    else:
+        assert False, "These types are not comparable"
+
+
+def test_repr():
+    ps = PhotoSize(TG, DICT_1)
+    assert repr(ps) == '<PhotoSize 12345>'
diff --git a/tests/unit/models/main_models/test_sticker.py b/tests/unit/models/main_models/test_sticker.py
@@ -27,3 +27,8 @@ def test_mask_pos():
         pass
     else:
         assert False
+
+
+def test_repr():
+    sticker = Sticker(TG, data=dict(width=512, height=510, file_id=12345))
+    assert repr(sticker) == '<Sticker 12345>'