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'])