#!/usr/bin/env python
# coding=utf-8

try:
    import mock

except ImportError:
    import unittest.mock as mock

import nltk
from marvin_nlp_ner_engine.data_handler import AcquisitorAndCleaner


@mock.patch('marvin_nlp_ner_engine.data_handler.acquisitor_and_cleaner.nltk.download')
@mock.patch('marvin_nlp_ner_engine.data_handler.acquisitor_and_cleaner.nltk.corpus')
def test_execute(corpus_mocked, download_mocked, mocked_params):

    ac = AcquisitorAndCleaner()
    ac.execute(params=mocked_params)

    download_mocked.assert_called_with('conll2002')
    corpus_mocked.conll2002.iob_sents.assert_called_with('esp.testb')