diff --git a/test/test_plugin/test_stub.py b/test/test_plugin/test_stub.py
index 8f992b97..691ac78f 100644
--- a/test/test_plugin/test_stub.py
+++ b/test/test_plugin/test_stub.py
@@ -24,3 +24,12 @@ def test_get_typeshed_directories():
     dirs = get_dirs(PythonVersionInfo(3, 6))
     assert dirs == {'stdlib/2and3', 'stdlib/3', 'stdlib/3.5', 'stdlib/3.6',
                     'third_party/2and3', 'third_party/3', 'third_party/3.5', 'third_party/3.6'}
+
+
+def test_get_stub_files():
+    def get_map(version_info):
+        return typeshed._create_stub_map(version_info)
+
+    base = os.path.join(typeshed._TYPESHED_PATH, 'stdlib', '3')
+    map_ = typeshed._create_stub_map(base)
+    assert map_['functools'] == os.path.join(base, 'functools.pyi')