Reformat using black

Author: mgautierfr

libzim/writer.py

@@ -132,9 +132,7 @@ class Creator:
 
     def __init__(self, filename, main_page, index_language, min_chunk_size):
         print(filename)
-        self._creatorWrapper = libzim_wrapper.Creator(
-            filename, main_page, index_language, min_chunk_size
-        )
+        self._creatorWrapper = libzim_wrapper.Creator(filename, main_page, index_language, min_chunk_size)
         self.filename = filename
         self.main_page = main_page
         self.language = index_language

tests/test_libzim.py

@@ -104,10 +104,7 @@ def article(article_content):
 
 def test_write_article(tmpdir, article):
     with Creator(
-        str(tmpdir / "test.zim"),
-        main_page="welcome",
-        index_language="eng",
-        min_chunk_size=2048,
+        str(tmpdir / "test.zim"), main_page="welcome", index_language="eng", min_chunk_size=2048,
     ) as zim_creator:
         zim_creator.add_article(article)
         zim_creator.update_metadata(
@@ -121,21 +118,15 @@ def test_write_article(tmpdir, article):
 
 def test_article_metadata(tmpdir, metadata):
     with Creator(
-        str(tmpdir / "test.zim"),
-        main_page="welcome",
-        index_language="eng",
-        min_chunk_size=2048,
+        str(tmpdir / "test.zim"), main_page="welcome", index_language="eng", min_chunk_size=2048,
     ) as zim_creator:
         zim_creator.update_metadata(**metadata)
         assert zim_creator._metadata == metadata
 
 
 def test_check_mandatory_metadata(tmpdir):
     with Creator(
-        str(tmpdir / "test.zim"),
-        main_page="welcome",
-        index_language="eng",
-        min_chunk_size=2048,
+        str(tmpdir / "test.zim"), main_page="welcome", index_language="eng", min_chunk_size=2048,
     ) as zim_creator:
         assert not zim_creator.mandatory_metadata_ok()
         zim_creator.update_metadata(

tests/test_libzim_file_reader.py

@@ -10,62 +10,65 @@
 
 ZIMFILES = [
     {
-        'filename': str(DATA_DIR/"wikipedia_es_physics_mini.zim"),
-        'checksum': u"99ea7a5598c6040c4f50b8ac0653b703",
-        'namespaces': u"-AIMX",
-        'article_count': 22027,
-        'main_page_url': u"A/index",
+        "filename": str(DATA_DIR / "wikipedia_es_physics_mini.zim"),
+        "checksum": "99ea7a5598c6040c4f50b8ac0653b703",
+        "namespaces": "-AIMX",
+        "article_count": 22027,
+        "main_page_url": "A/index",
     }
 ]
 
 
-
-
 @pytest.fixture(params=ZIMFILES)
 def zimdata(request):
     return request.param
 
+
 @pytest.fixture
 def reader(zimdata):
-    return File(zimdata['filename'])
+    return File(zimdata["filename"])
 
 
 @pytest.fixture
 def article_data():
     return {
-        'url': u"A/Albert_Einstein",
-        'title': u"Albert Einstein",
-        'mimetype':u"text/html",
-        'article_id': 663,
-        'size': 17343
+        "url": "A/Albert_Einstein",
+        "title": "Albert Einstein",
+        "mimetype": "text/html",
+        "article_id": 663,
+        "size": 17343,
     }
 
 
 def test_zim_filename(reader, zimdata):
     for k, v in zimdata.items():
         assert getattr(reader, k) == v
 
+
 def test_zim_read(reader, article_data):
-    article = reader.get_article(article_data['url'])
+    article = reader.get_article(article_data["url"])
 
-    assert article.longurl == article_data['url']
-    assert article.title == article_data['title']
-    assert article.url == article_data['url'][2:]
-    assert article.mimetype == article_data['mimetype']
+    assert article.longurl == article_data["url"]
+    assert article.title == article_data["title"]
+    assert article.url == article_data["url"][2:]
+    assert article.mimetype == article_data["mimetype"]
     assert isinstance(article.content, memoryview)
-    assert len(article.content) == article_data['size']
+    assert len(article.content) == article_data["size"]
+
 
 def test_content_ref_keep(reader):
     """Get the memoryview on a content and loose the reference on the article.
        We try to load a lot of other articles to detect possible use of dandling pointer
     """
-    content =None
+    content = None
+
     def get_content():
         nonlocal content
-        article = reader.get_article(u"A/Albert_Einstein")
+        article = reader.get_article("A/Albert_Einstein")
         assert isinstance(article.content, memoryview)
         content = article.content
-    get_content() # Now we have a content but no reference to the article.
+
+    get_content()  # Now we have a content but no reference to the article.
     gc.collect()
     # Load a lot of content
     for i in range(0, reader.article_count, 2):
@@ -74,28 +77,35 @@ def get_content():
             c = article.content
     # Check everything is ok
     assert len(content) == 17343
-    assert bytes(content[:100]) == b'<!DOCTYPE html>\n<html class="client-js"><head>\n  <meta charset="UTF-8">\n  <title>Albert Einstein</ti'
+    assert (
+        bytes(content[:100])
+        == b'<!DOCTYPE html>\n<html class="client-js"><head>\n  <meta charset="UTF-8">\n  <title>Albert Einstein</ti'
+    )
+
 
 def test_get_article_by_id(reader, article_data):
     return
-    article = reader.get_article_by_id(article_data['article_id'])
+    article = reader.get_article_by_id(article_data["article_id"])
+
+    assert article.longurl == article_data["url"]
+    assert article.title == article_data["title"]
+    assert article.url == article_data["url"][2:]
+    assert article.mimetype == article_data["mimetype"]
 
-    assert article.longurl == article_data['url']
-    assert article.title == article_data['title']
-    assert article.url == article_data['url'][2:]
-    assert article.mimetype == article_data['mimetype']
 
 def test_namespace_count(reader):
     namespaces = reader.namespaces
     num_articles = sum(reader.get_namespaces_count(ns) for ns in namespaces)
     assert reader.article_count == num_articles
 
+
 def test_suggest(reader):
-    results =  reader.suggest(u"Einstein")
-    assert u"A/Albert_Einstein" in list(results)
+    results = reader.suggest("Einstein")
+    assert "A/Albert_Einstein" in list(results)
+
 
 def test_search(reader):
-    results = reader.search(u"Einstein")
+    results = reader.search("Einstein")
     assert len(list(results)) == 10