1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
# -*- coding: utf-8 -*-
"""Tests for PyBEL utilities."""
import unittest
from pybel.exceptions import PlaceholderAminoAcidWarning
from pybel.parser.modifiers.constants import amino_acid
from pybel.parser.utils import nest
from pybel.utils import expand_dict, flatten_dict, tokenize_version
class TestTokenizeVersion(unittest.TestCase):
"""Test tokenization of version strings."""
def test_simple(self):
"""Test the simplest version string case."""
version_str = "0.1.2"
version_tuple = 0, 1, 2
self.assertEqual(version_tuple, tokenize_version(version_str))
def test_long(self):
"""Test when the version pieces have more than 1 digit."""
version_str = "0.12.20"
version_tuple = 0, 12, 20
self.assertEqual(version_tuple, tokenize_version(version_str))
def test_dev(self):
"""Test when there's a dash after."""
version_str = "0.1.2-dev"
version_tuple = 0, 1, 2
self.assertEqual(version_tuple, tokenize_version(version_str))
class TestRandom(unittest.TestCase):
def test_nest_failure(self):
with self.assertRaises(ValueError):
nest()
def test_bad_aminoAcid(self):
with self.assertRaises(PlaceholderAminoAcidWarning):
amino_acid.parseString("X")
class TestUtils(unittest.TestCase):
def test_expand_dict(self):
flat_dict = {
"k1": "v1",
"k2_k2a": "v2",
"k2_k2b": "v3",
"k2_k2c_k2ci": "v4",
"k2_k2c_k2cii": "v5",
}
expected_dict = {
"k1": "v1",
"k2": {"k2a": "v2", "k2b": "v3", "k2c": {"k2ci": "v4", "k2cii": "v5"}},
}
self.assertEqual(expected_dict, expand_dict(flat_dict))
def test_flatten_dict(self):
d = {"A": 5, "B": "b", "C": {"D": "d", "E": "e"}}
expected = {"A": 5, "B": "b", "C_D": "d", "C_E": "e"}
self.assertEqual(expected, flatten_dict(d))
def test_flatten_dict_withLists(self):
d = {"A": 5, "B": "b", "C": {"D": ["d", "delta"], "E": "e"}}
expected = {"A": 5, "B": "b", "C_D": "d,delta", "C_E": "e"}
self.assertEqual(expected, flatten_dict(d))
|
