File: test_node_data.py

package info (click to toggle)
augur 24.4.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 25,312 kB
  • sloc: python: 14,253; sh: 227; makefile: 35
file content (34 lines) | stat: -rw-r--r-- 1,123 bytes parent folder | download | duplicates (4)
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
from augur.util_support.node_data import DuplicatedNonDictAttributeError
from augur.util_support.node_data import NodeData

import pytest


class TestNodeData:
    def test_update(self):
        node_data = NodeData()
        node_data.update({"a": 5, "b": {"c": 6}})
        assert node_data.attrs == {"a": 5, "b": {"c": 6}}

        node_data.update({"a": 7})
        assert node_data.attrs == {"a": 7, "b": {"c": 6}}

        node_data.update({"b": {"d": 8}})
        assert node_data.attrs == {"a": 7, "b": {"c": 6, "d": 8}}

        node_data.update({"b": {"d": 9}})
        assert node_data.attrs == {"a": 7, "b": {"c": 6, "d": 9}}

    def test_update_merge_nondict_into_dict(self):
        node_data = NodeData()
        node_data.update({"a": 5, "b": {"c": 6}})

        with pytest.raises(DuplicatedNonDictAttributeError):
            node_data.update({"b": "a string!"})

    def test_update_merge_dict_into_nondict(self):
        node_data = NodeData()
        node_data.update({"a": 5, "b": {"c": 6}})

        with pytest.raises(DuplicatedNonDictAttributeError):
            node_data.update({"a": {"g": 2}})