def main(): if user_set_encoding_and_is_utf8(): check_zdict_dir_and_db() global dictionary_map dictionary_map = get_dictionary_map() args = get_args() args = set_args(args) execute_zdict(args) else: exit()
def test_get_dictionary_map(listdir, import_module, getmembers): # prepare mocked return value MockDict = MagicMock(name='MockDict') MockDict().provider = 'mock' listdir.return_value = [ '__init__.py', '_strange.py', 'mock.py', 'template.py', # the excluded files 'not_a_py.rst', ] getmembers.return_value = [('mock', MockDict)] dict_map = get_dictionary_map() assert dict_map == {'mock': MockDict} assert listdir.called import_module.assert_called_with('zdict.dictionaries.mock') assert import_module.call_count == 1 getmembers.assert_called_with('mock', predicate=_is_dict) assert MockDict.called