def test_get_filter_names(self):
mock_filter_cls = MagicMock()
mock_filter_cls.BOOLEAN_FILTERS = {'foo': None, 'bar': None}
mock_filter_cls.COMPARATIVE_FILTERS = {'baz': None}
self.assertEqual(
tuple(candidates._utils.get_filter_names(mock_filter_cls)),
('foo', 'bar', 'baz'))
def test_get_filter_spec(self):
mock_filter_cls = MagicMock()
mock_filter_cls.BOOLEAN_FILTERS = {'foo': 'mock foo spec', 'bar': 'mock bar spec'}
mock_filter_cls.COMPARATIVE_FILTERS = {'baz': 'mock baz spec'}
self.assertEqual(candidates._get_filter_spec(mock_filter_cls, 'foo'), 'mock foo spec')
self.assertEqual(candidates._get_filter_spec(mock_filter_cls, 'bar'), 'mock bar spec')
self.assertEqual(candidates._get_filter_spec(mock_filter_cls, 'baz'), 'mock baz spec')
def test_filter_takes_completable_values(self):
mock_filter_cls = MagicMock()
mock_filter_cls.BOOLEAN_FILTERS = {'foo': None}
mock_filter_cls.COMPARATIVE_FILTERS = {'bar': SimpleNamespace(value_type=str),
'baz': SimpleNamespace(value_type=int)}
self.assertEqual(candidates._filter_takes_completable_values(mock_filter_cls, 'bar'), True)
self.assertEqual(candidates._filter_takes_completable_values(mock_filter_cls, 'baz'), False)
self.assertEqual(candidates._filter_takes_completable_values(mock_filter_cls, 'foo'), False)
self.assertEqual(candidates._filter_takes_completable_values(mock_filter_cls, 'asdf'), False)
def test_get_filter_names(self):
mock_filter_cls = MagicMock()
mock_filter_cls.BOOLEAN_FILTERS = {'foo': None, 'bar': None}
mock_filter_cls.COMPARATIVE_FILTERS = {'baz': None}
import sys
if sys.hexversion < 0x03060000:
# Python <= 3.6 dicts are not ordered yet
self.assertEqual(
set(candidates._utils.get_filter_names(mock_filter_cls)),
{'foo', 'bar', 'baz'})
else:
self.assertEqual(
tuple(candidates._utils.get_filter_names(mock_filter_cls)),
('foo', 'bar', 'baz'))
