Skip to content

Commit 8ccce80

Browse files
author
Thomas Basche
committed
Add tests for data formatting functions
1 parent 2f8238b commit 8ccce80

File tree

2 files changed

+70
-0
lines changed

2 files changed

+70
-0
lines changed

reposit_tests/test_deepest_key.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
import pytest
2+
3+
from reposit.data.utils import deepest_key
4+
5+
three_levels = {
6+
'level1': {
7+
'level2': {
8+
'level3': {}
9+
}
10+
}
11+
}
12+
13+
single_level = {
14+
'level1': {}
15+
}
16+
17+
two_levels = {
18+
'level1': {
19+
'level2': {}
20+
}
21+
}
22+
23+
24+
@pytest.mark.parametrize('_dict, expected', [
25+
(single_level, 'level1'),
26+
(two_levels, 'level2'),
27+
(three_levels, 'level3'),
28+
])
29+
def test_deepest_key_func(_dict, expected):
30+
assert deepest_key(_dict) == expected

reposit_tests/test_match_schema.py

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
import pytest
2+
3+
from reposit.data.utils import match_to_schema
4+
5+
key_1 = 'inverterP'
6+
7+
data_1 = {
8+
'data': {
9+
'inverterP': 40.0
10+
}
11+
}
12+
13+
key_2 = 'houseP'
14+
15+
data_2 = {
16+
'houseP': 50.00
17+
}
18+
19+
key_3 = 'blah2'
20+
21+
data_3 = {
22+
'data': {
23+
'generation': {
24+
'some_data': {
25+
'blah': 2,
26+
'blah2': 1,
27+
'blah3': 40,
28+
}
29+
}
30+
}
31+
}
32+
33+
34+
@pytest.mark.parametrize('_dict, requested_key, expected', [
35+
(data_1, key_1, 40.0),
36+
(data_2, key_2, 50.0),
37+
(data_3, key_3, 1),
38+
])
39+
def test_match_schema_func(_dict, requested_key, expected):
40+
assert match_to_schema(_dict, requested_key) == expected

0 commit comments

Comments
 (0)