python-toolbox/marvin_python_toolbox/management/templates/python-engine/tests/data_handler/test_training_preparator.py

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

try:
    import mock

except ImportError:
    import unittest.mock as mock

from {{project.package}}.data_handler import TrainingPreparator


class TestTrainingPreparator:
    def test_execute(self, mocked_params):
        ac = TrainingPreparator()
        ac.execute(params=mocked_params)
        assert not ac._params