chore: add test for reading from file

Author: corneliusroemer

tests/test_aliasor.py

@@ -1,4 +1,26 @@
+from __future__ import unicode_literals
+
+import os
+from distutils import dir_util
+
 from pango_aliasor.aliasor import Aliasor
+from pytest import fixture
+
+
+@fixture
+def datadir(tmpdir, request):
+    '''
+    Fixture responsible for searching a folder with the same name of test
+    module and, if available, moving all contents to a temporary directory so
+    tests can use them freely.
+    '''
+    filename = request.module.__file__
+    test_dir, _ = os.path.splitext(filename)
+
+    if os.path.isdir(test_dir):
+        dir_util.copy_tree(test_dir,str(tmpdir))
+
+    return tmpdir
 
 def test_pango_designation_init():
     aliasor = Aliasor()
@@ -33,4 +55,8 @@ def test_double_alias_compression():
 
 def test_double_alias_uncompression():
     aliasor = Aliasor()
-    assert aliasor.uncompress('BE.1') == 'B.1.1.529.5.3.1.1'
+    assert aliasor.uncompress('BE.1') == 'B.1.1.529.5.3.1.1'
+
+def test_read_from_file(datadir):
+    aliasor = Aliasor(datadir.join('alias_key.json'))
+    assert aliasor.compress('B.1.1.529.1') == 'BA.1'

tests/test_aliasor/alias_key.json

@@ -0,0 +1,80 @@
+{   
+    "A": "",
+    "B": "",
+    "C": "B.1.1.1",
+    "D": "B.1.1.25",
+    "G": "B.1.258.2",
+    "K": "B.1.1.277",
+    "L": "B.1.1.10",
+    "M": "B.1.1.294",
+    "N": "B.1.1.33",
+    "P": "B.1.1.28",
+    "Q": "B.1.1.7",
+    "R": "B.1.1.316",
+    "S": "B.1.1.217",
+    "U": "B.1.177.60",
+    "V": "B.1.177.54",
+    "W": "B.1.177.53",
+    "Y": "B.1.177.52",
+    "Z": "B.1.177.50",
+    "AA": "B.1.177.15",
+    "AB": "B.1.160.16",
+    "AC": "B.1.1.405",
+    "AD": "B.1.1.315",
+    "AE": "B.1.1.306",
+    "AF": "B.1.1.305",
+    "AG": "B.1.1.297",
+    "AH": "B.1.1.241",
+    "AJ": "B.1.1.240",
+    "AK": "B.1.1.232",
+    "AL": "B.1.1.231",
+    "AM": "B.1.1.216",
+    "AN": "B.1.1.200",
+    "AP": "B.1.1.70",
+    "AQ": "B.1.1.39",
+    "AS": "B.1.1.317",
+    "AT": "B.1.1.370",
+    "AU": "B.1.466.2",
+    "AV": "B.1.1.482",
+    "AW": "B.1.1.464",
+    "AY": "B.1.617.2",
+    "AZ": "B.1.1.318",
+    "BA": "B.1.1.529",
+    "BB": "B.1.621.1",
+    "BC": "B.1.1.529.1.1.1",
+    "BD": "B.1.1.529.1.17.2",
+    "BE": "B.1.1.529.5.3.1",
+    "BF": "B.1.1.529.5.2.1",
+    "BG": "B.1.1.529.2.12.1",
+    "XA": ["B.1.1.7","B.1.177"],
+    "XB": ["B.1.634","B.1.631"],
+    "XC": ["AY.29","B.1.1.7"],
+    "XD": ["B.1.617.2*","BA.1*"],
+    "XE": ["BA.1*","BA.2*"],
+    "XF": ["B.1.617.2*","BA.1*"],
+    "XG": ["BA.1*","BA.2*"],
+    "XH": ["BA.1*","BA.2*"],
+    "XJ": ["BA.1*","BA.2*"],
+    "XK": ["BA.1*","BA.2*"],
+    "XL": ["BA.1*","BA.2*"],
+    "XM": ["BA.1.1*","BA.2*"],
+    "XN": ["BA.1*","BA.2*"],
+    "XP": ["BA.1.1*","BA.2*"],
+    "XQ": ["BA.1.1*","BA.2*"],
+    "XR": ["BA.1.1*","BA.2*"],
+    "XS": ["B.1.617.2*","BA.1.1*"],
+    "XT": ["BA.2*","BA.1*"],
+    "XU": ["BA.1*","BA.2*"],
+    "XV": ["BA.1*","BA.2*"],
+    "XW": ["BA.1*","BA.2*"],
+    "XY": ["BA.1*","BA.2*"],
+    "XZ": ["BA.2*","BA.1*"],
+    "XAA": ["BA.1*","BA.2*"],
+    "XAB": ["BA.1*","BA.2*"],
+    "XAC": ["BA.2*","BA.1*","BA.2*"],
+    "XAD": ["BA.2*","BA.1*"],
+    "XAE": ["BA.2*","BA.1*"],
+    "XAF": ["BA.1*","BA.2*"],
+    "XAG": ["BA.1*","BA.2*"],
+    "XAH": ["BA.2*","BA.1*"]
+}