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'))