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