bazarr/libs/rebulk/test/test_rules.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# pylint: disable=no-self-use, pointless-statement, missing-docstring, invalid-name, no-member, len-as-condition
import pytest
from rebulk.test.default_rules_module import RuleRemove0, RuleAppend0, RuleRename0, RuleAppend1, RuleRemove1, \
    RuleRename1, RuleAppend2, RuleRename2, RuleAppend3, RuleRename3, RuleAppendTags0, RuleRemoveTags0, \
    RuleAppendTags1, RuleRemoveTags1

from ..rules import Rules
from ..match import Matches, Match

from .rules_module import Rule1, Rule2, Rule3, Rule0, Rule1Disabled
from . import rules_module as rm


def test_rule_priority():
    matches = Matches([Match(1, 2)])

    rules = Rules(Rule1, Rule2())

    rules.execute_all_rules(matches, {})
    assert len(matches) == 0
    matches = Matches([Match(1, 2)])

    rules = Rules(Rule1(), Rule0)

    rules.execute_all_rules(matches, {})
    assert len(matches) == 1
    assert matches[0] == Match(3, 4)


def test_rules_duplicates():
    matches = Matches([Match(1, 2)])

    rules = Rules(Rule1, Rule1)

    with pytest.raises(ValueError):
        rules.execute_all_rules(matches, {})


def test_rule_disabled():
    matches = Matches([Match(1, 2)])

    rules = Rules(Rule1Disabled(), Rule2())

    rules.execute_all_rules(matches, {})
    assert len(matches) == 2
    assert matches[0] == Match(1, 2)
    assert matches[1] == Match(3, 4)


def test_rule_when():
    matches = Matches([Match(1, 2)])

    rules = Rules(Rule3())

    rules.execute_all_rules(matches, {'when': False})
    assert len(matches) == 1
    assert matches[0] == Match(1, 2)

    matches = Matches([Match(1, 2)])

    rules.execute_all_rules(matches, {'when': True})
    assert len(matches) == 2
    assert matches[0] == Match(1, 2)
    assert matches[1] == Match(3, 4)


class TestDefaultRules(object):
    def test_remove(self):
        rules = Rules(RuleRemove0)

        matches = Matches([Match(1, 2)])
        rules.execute_all_rules(matches, {})

        assert len(matches) == 0

        rules = Rules(RuleRemove1)

        matches = Matches([Match(1, 2)])
        rules.execute_all_rules(matches, {})

        assert len(matches) == 0

    def test_append(self):
        rules = Rules(RuleAppend0)

        matches = Matches([Match(1, 2)])
        rules.execute_all_rules(matches, {})

        assert len(matches) == 2

        rules = Rules(RuleAppend1)

        matches = Matches([Match(1, 2)])
        rules.execute_all_rules(matches, {})

        assert len(matches) == 2

        rules = Rules(RuleAppend2)

        matches = Matches([Match(1, 2)])
        rules.execute_all_rules(matches, {})

        assert len(matches) == 2
        assert len(matches.named('renamed')) == 1

        rules = Rules(RuleAppend3)

        matches = Matches([Match(1, 2)])
        rules.execute_all_rules(matches, {})

        assert len(matches) == 2
        assert len(matches.named('renamed')) == 1

    def test_rename(self):
        rules = Rules(RuleRename0)

        matches = Matches([Match(1, 2, name='original')])
        rules.execute_all_rules(matches, {})

        assert len(matches.named('original')) == 1
        assert len(matches.named('renamed')) == 0

        rules = Rules(RuleRename1)

        matches = Matches([Match(5, 10, name='original')])
        rules.execute_all_rules(matches, {})

        assert len(matches.named('original')) == 0
        assert len(matches.named('renamed')) == 1

        rules = Rules(RuleRename2)

        matches = Matches([Match(5, 10, name='original')])
        rules.execute_all_rules(matches, {})

        assert len(matches.named('original')) == 0
        assert len(matches.named('renamed')) == 1

        rules = Rules(RuleRename3)

        matches = Matches([Match(5, 10, name='original')])
        rules.execute_all_rules(matches, {})

        assert len(matches.named('original')) == 0
        assert len(matches.named('renamed')) == 1

    def test_append_tags(self):
        rules = Rules(RuleAppendTags0)

        matches = Matches([Match(1, 2, name='tags', tags=['other'])])
        rules.execute_all_rules(matches, {})

        assert len(matches.named('tags')) == 1
        assert matches.named('tags', index=0).tags == ['other', 'new-tag']

        rules = Rules(RuleAppendTags1)

        matches = Matches([Match(1, 2, name='tags', tags=['other'])])
        rules.execute_all_rules(matches, {})

        assert len(matches.named('tags')) == 1
        assert matches.named('tags', index=0).tags == ['other', 'new-tag']

    def test_remove_tags(self):
        rules = Rules(RuleRemoveTags0)

        matches = Matches([Match(1, 2, name='tags', tags=['other', 'new-tag'])])
        rules.execute_all_rules(matches, {})

        assert len(matches.named('tags')) == 1
        assert matches.named('tags', index=0).tags == ['other']

        rules = Rules(RuleRemoveTags1)

        matches = Matches([Match(1, 2, name='tags', tags=['other', 'new-tag'])])
        rules.execute_all_rules(matches, {})

        assert len(matches.named('tags')) == 1
        assert matches.named('tags', index=0).tags == ['other']


def test_rule_module():
    rules = Rules(rm)

    matches = Matches([Match(1, 2)])
    rules.execute_all_rules(matches, {})

    assert len(matches) == 1


def test_rule_repr():
    assert str(Rule0()) == "<Rule0>"
    assert str(Rule1()) == "<Rule1>"
    assert str(Rule2()) == "<Rule2>"
    assert str(Rule1Disabled()) == "<Disabled Rule1>"