X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/12b5230d8fdb3ad995e867fb5d58a69e8a627e68..deca3e31b94bc7ec95d47c309ff560a097337e97:/tests/test_html_fragments.py diff --git a/tests/test_html_fragments.py b/tests/test_html_fragments.py index 3e87a9e..b3915fd 100644 --- a/tests/test_html_fragments.py +++ b/tests/test_html_fragments.py @@ -3,16 +3,21 @@ # This file is part of Librarian, licensed under GNU Affero GPLv3 or later. # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. # +from __future__ import unicode_literals + +import unittest from librarian.html import extract_fragments -from nose.tools import * -from utils import get_fixture +from .utils import get_fixture + +class FragmentsTest(unittest.TestCase): + maxDiff = None -def test_fragments(): - expected_output_file_path = get_fixture('text', 'asnyk_miedzy_nami_fragments.html') + def test_fragments(self): + expected_output_file_path = get_fixture('text', 'asnyk_miedzy_nami_fragments.html') - closed_fragments, open_fragments = extract_fragments( - get_fixture('text', 'asnyk_miedzy_nami_expected.html')) - assert not open_fragments - fragments_text = u"\n\n".join(u"%s: %s\n%s" % (f.id, f.themes, f) for f in closed_fragments.values()) - assert_equal(fragments_text, file(expected_output_file_path).read().decode('utf-8')) + closed_fragments, open_fragments = extract_fragments( + get_fixture('text', 'asnyk_miedzy_nami_expected.html')) + assert not open_fragments + fragments_text = u"\n\n".join(u"%s: %s\n%s" % (f.id, f.themes, f) for f in sorted(closed_fragments.values(), key=lambda f: f.id)) + self.assertEqual(fragments_text, open(expected_output_file_path, 'rb').read().decode('utf-8'))