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')
def test_cache_train(self, exists_mock, load_mock, open_mock, calc_mock):
categories = BayesCategories()
categories.categories = {'foo': 'bar'}
load_mock.return_value = categories
open_mock.return_value = 'opened'
exists_mock.return_value = True
sb = SimpleBayes(cache_path='foo')
sb.cache_train()
exists_mock.assert_called_once_with('foo/_simplebayes.pickle')
open_mock.assert_called_once_with('foo/_simplebayes.pickle', 'rb')
load_mock.assert_called_once_with('opened')
calc_mock.assert_called_once_with()
self.assertEqual(sb.categories, categories)
def test_cache_train(self, exists_mock, load_mock, open_mock, calc_mock):
categories = BayesCategories()
categories.categories = {'foo': 'bar'}
load_mock.return_value = categories
open_mock.return_value = 'opened'
exists_mock.return_value = True
sb = SimpleBayes(cache_path='foo')
sb.cache_train()
exists_mock.assert_called_once_with('foo/_simplebayes.pickle')
open_mock.assert_called_once_with('foo/_simplebayes.pickle', 'rb')
load_mock.assert_called_once_with('opened')
calc_mock.assert_called_once_with()
self.assertEqual(sb.categories, categories)