from deidentify.surrogates.generators import EmailSurrogates

from .util import RandomDataMock


def test_replace_all():
    annotations = [
        '<EMAIL>',
        '<EMAIL>',
        '<EMAIL>'
    ]

    email_surrogates = EmailSurrogates(annotations=annotations, random_data=RandomDataMock())

    assert email_surrogates.replace_all() == [
        '<EMAIL>',
        '<EMAIL>',
        '<EMAIL>'
    ]

from deidentify.surrogates.generators import EmailSurrogates

from .util import RandomDataMock


def test_replace_all():
    annotations = [
        '<EMAIL>',
        '<EMAIL>',
        '<EMAIL>'
    ]

    email_surrogates = EmailSurrogates(annotations=annotations, random_data=RandomDataMock())

    assert email_surrogates.replace_all() == [
        '<EMAIL>',
        '<EMAIL>',
        '<EMAIL>'
    ]
