Tests (inspired by #8)

Author: VictorLeP

setup.py

@@ -20,5 +20,11 @@
     ],
     install_requires=[
         'requests'
-    ]
+    ],
+    extras_require={
+        'dev': [
+            'pytest',
+            'codetiming'
+        ]
+    },
 )

tests/__init__.py

@@ -0,0 +1,15 @@
+import pytest
+from tranco.tranco import Tranco
+
+
+@pytest.fixture(scope="session")
+def tranco():
+    t = Tranco(cache=True, cache_dir='.tranco')
+    t.list()  # prefetch list
+    return t
+
+
+@pytest.fixture(scope="session")
+def tranco_list(tranco):
+    return tranco.list()
+

tests/conftest.py

@@ -0,0 +1,18 @@
+import uuid
+
+
+def test_top_10(tranco_list):
+    assert len(tranco_list.top(10)) == 10
+
+
+def test_top_1000000(tranco_list):
+    assert len(tranco_list.top(1000000)) == 1000000
+
+
+def test_domain_rank(tranco_list):
+    top_1 = tranco_list.top(1)[0]
+    assert tranco_list.rank(top_1) == 1
+
+
+def test_domain_not_in_list_rank(tranco_list):
+    assert tranco_list.rank(f"domaindoesntexist{uuid.uuid4().hex.upper()[0:6]}.com") == -1

tests/test_list.py

@@ -0,0 +1,20 @@
+from codetiming import Timer
+import pytest
+
+
+def test_performance_rank(tranco_list):
+    core_list = tranco_list.top(1000000)     
+    # test dictionary
+    timer = Timer(name="class", logger=None)
+    timer.start()
+    for i, domain in enumerate(core_list, start=1):
+        assert i == tranco_list.rank(domain)
+    print("Rank:", timer.stop())
+
+
+def test_performance_top(tranco_list):
+    # test dictionary
+    timer = Timer(name="class", logger=None)
+    timer.start()
+    core_list = tranco_list.top(1000000)     
+    print("Top:", timer.stop())