diff --git a/tests/images/solution-pokemon_jakt.png b/tests/images/solution-pokemon_jakt.png new file mode 100644 index 0000000..9304531 Binary files /dev/null and b/tests/images/solution-pokemon_jakt.png differ diff --git a/tests/test_luke04.py b/tests/test_luke04.py index 376886c..0e67e21 100644 --- a/tests/test_luke04.py +++ b/tests/test_luke04.py @@ -1,2 +1,12 @@ +import os +from PIL import Image +from luke_04.LSB import main + +path = os.path.dirname(__file__) + def test_answer(): - assert False + solution_path = os.path.join(path, 'images/solution-pokemon_jakt.png') + solution_image = Image.open(solution_path) + answer_image = Image.open(main()) + + assert solution_image.tobytes == answer_image.tobytes