dimanche 18 décembre 2016

Remove mock tests

I am currently using mock library for testing and everything is running fine. But now I need to remove mock from the tests. Is there any solution to do so?

Here is a snippet of my code:

from mock import patch, call

@patch("pgmpy.models.MarkovChain._check_state", autospec=True)
def test_set_start_state_list(self, check_state):
    model = MC(['b', 'a'], [1, 2])
    check_state.return_value = True
    model.set_start_state([State('a', 0), State('b', 1)])
    model_state = [State('b', 1), State('a', 0)]
    check_state.assert_called_once_with(model, model_state)
    self.assertEqual(model.state, model_state)

def test_set_start_state_none(self):
    model = MC()
    model.state = 'state'
    model.set_start_state(None)
    self.assertIsNone(model.state)

Aucun commentaire:

Enregistrer un commentaire