def test_persist_cache(self, dump_mock, open_mock):
open_mock.return_value = 'opened'
categories = BayesCategories()
categories.categories = {'foo': 'bar'}
sb = SimpleBayes()
sb.cache_path = '/tmp/'
sb.categories = categories
sb.cache_persist()
open_mock.assert_called_once_with('/tmp/_simplebayes.pickle', 'wb')
dump_mock.assert_called_once_with(categories, 'opened')
def test_persist_cache(self, dump_mock, open_mock):
open_mock.return_value = 'opened'
categories = BayesCategories()
categories.categories = {'foo': 'bar'}
sb = SimpleBayes()
sb.cache_path = '/tmp/'
sb.categories = categories
sb.cache_persist()
open_mock.assert_called_once_with('/tmp/_simplebayes.pickle', 'wb')
dump_mock.assert_called_once_with(categories, 'opened')
