From 4f0d86be186c290b199ee69b1f929c24de59c556 Mon Sep 17 00:00:00 2001 From: Cody Logan Date: Tue, 7 Nov 2023 11:07:10 -0800 Subject: Add tests for File equality --- tests/test_file_class.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/test_file_class.py b/tests/test_file_class.py index ede4be1..42f5667 100644 --- a/tests/test_file_class.py +++ b/tests/test_file_class.py @@ -60,3 +60,12 @@ class TestFileClass: """ file = File("foobar.jpg", site="en.wikipedia.org") assert file.site == "en.wikipedia.org" + + def test_file_equality(self, file_with_name: File) -> None: + assert File(name="foobar.jpg") == file_with_name + + def test_file_inequality(self, file_with_name: File) -> None: + assert File(name="foobaz.jpg", dest="output.jpg") != file_with_name + + def test_file_comparison_with_non_file(self, file_with_name: File) -> None: + assert file_with_name.__eq__({"name": "foobar.jpg"}) == NotImplemented -- cgit v1.2.3