def test_merge_nodes():
left_char = CharNode("A", 13)
right_char = CharNode("B", 27)
new_node = left_char + right_char
assert new_node.value == "AB"
assert new_node.freq == left_char.freq + right_char.freq
assert new_node.left is left_char
assert new_node.right is right_char
def process_frequencies(stream: Sequence[str]) -> List[CharNode]:
"""
Given a stream of text, return a list of CharNode with the frequencies
for each character.
:param stream: sequence with all the characters.
"""
counts = Counter(stream)
return [CharNode(value=value, freq=freq) for value, freq in counts.items()]
def test_merge_invalid_nodes():
with pytest.raises(TypeError):
CharNode("A", 1) + "B"
def test_charnode_eq():
assert CharNode("a", 42) == CharNode("a", 42)
def test_charnode_is_not_leaf():
left = CharNode("left", 1)
right = CharNode("right", 1)
root = CharNode("root", 1, left, right)
assert root.leaf is False
def test_charnode_is_leaf():
assert CharNode("leaf", 1).leaf is True
def test_charnode_le_different_class():
with pytest.raises(TypeError):
assert CharNode("node", 1) <= 2
def test_charnode_le():
assert CharNode("lte", 1) <= CharNode("lte", 1)
import pytest
from compressor.char_node import CharNode
def test_charnode_eq():
assert CharNode("a", 42) == CharNode("a", 42)
@pytest.mark.parametrize(
"charnode,different",
((CharNode("node", 1), 1), (CharNode("first", 1), CharNode("second", 1))),
)
def test_charnode_eq_type_mismatch(charnode, different):
assert charnode != different
@pytest.mark.parametrize(
"smaller,bigger",
(
(CharNode("smaller", 1), CharNode("bigger", 2)),
(CharNode("a", 4), CharNode("a", 5)),
),
)
def test_charnode_lt(smaller, bigger):
assert smaller < bigger
def test_charnode_le():
assert CharNode("lte", 1) <= CharNode("lte", 1)