<reponame>yafeunteun/wikipedia-spam-classifier
import pickle
import re

from nose.tools import eq_

from .. import filters
from ....dependencies import solve
from ...datasource import Datasource

tokens = Datasource("tokens")

foo_tokens = filters.regex_matching("foo", tokens, name="foo_tokens")
foo_case_tokens = filters.regex_matching(re.compile("foo"), tokens,
                                         name="foo_case_tokens")

my_ints = Datasource("my_ints")

positive_ints = filters.positive(my_ints)
negative_ints = filters.negative(my_ints)


def test_regex_matching():
    cache = {tokens: ["foo", "bar", "FOO"]}
    eq_(solve(foo_tokens, cache=cache),
        ["foo", "FOO"])

    eq_(solve(foo_case_tokens, cache=cache),
        ["foo"])

    eq_(pickle.loads(pickle.dumps(foo_tokens)), foo_tokens)
    eq_(pickle.loads(pickle.dumps(foo_case_tokens)), foo_case_tokens)


def test_positive():
    cache = {my_ints: [1, 0, -1]}
    eq_(solve(positive_ints, cache=cache),
        [1])
    eq_(pickle.loads(pickle.dumps(positive_ints)), positive_ints)


def test_negative():
    cache = {my_ints: [1, 0, -1]}
    eq_(solve(negative_ints, cache=cache),
        [-1])
    eq_(pickle.loads(pickle.dumps(negative_ints)), negative_ints)

import pickle
import re

from nose.tools import eq_

from .. import filters
from ....dependencies import solve
from ...datasource import Datasource

tokens = Datasource("tokens")

foo_tokens = filters.regex_matching("foo", tokens, name="foo_tokens")
foo_case_tokens = filters.regex_matching(re.compile("foo"), tokens,
                                         name="foo_case_tokens")

my_ints = Datasource("my_ints")

positive_ints = filters.positive(my_ints)
negative_ints = filters.negative(my_ints)


def test_regex_matching():
    cache = {tokens: ["foo", "bar", "FOO"]}
    eq_(solve(foo_tokens, cache=cache),
        ["foo", "FOO"])

    eq_(solve(foo_case_tokens, cache=cache),
        ["foo"])

    eq_(pickle.loads(pickle.dumps(foo_tokens)), foo_tokens)
    eq_(pickle.loads(pickle.dumps(foo_case_tokens)), foo_case_tokens)


def test_positive():
    cache = {my_ints: [1, 0, -1]}
    eq_(solve(positive_ints, cache=cache),
        [1])
    eq_(pickle.loads(pickle.dumps(positive_ints)), positive_ints)


def test_negative():
    cache = {my_ints: [1, 0, -1]}
    eq_(solve(negative_ints, cache=cache),
        [-1])
    eq_(pickle.loads(pickle.dumps(negative_ints)), negative_ints)