def test_get_averaged_scorevariations():
node = Node('a', score=10)
node.variations = ['v1', 'v2']
assert node.get_averaged_score() == 5.0
def test_has_variation():
node = Node('a')
node.variations = ['v']
assert node.has_variation('v')
assert not node.has_variation('w')
def test_get_variations():
node = Node('a')
node.variations = ['v']
assert set(node.get_variations()) == set(['v'])
def test_remove_variation():
node = Node('a')
node.variations = ['v1', 'v2']
node.remove_variation('v1')
assert set(node.variations) == set(['v2'])
