Put version setting to a function

data_repo_renderer/__init__.py

@@ -9,12 +9,31 @@ import stat
 import textwrap
 import yaml
 
+
+def maybe_get_installed_version(pkg_name):
+    """If `pkg_name` is installed, set version from there.
+
+    Attributes
+    ----------
+    pkg_name : str
+        Package name.
+
+    Returns
+    -------
+    version : str
+        Version number from installed package or 'vX.X.X'
+
+    """
+    try:
+        version = get_distribution(pkg_name).version
+    except DistributionNotFound:
+        # package is not installed
+        version = "vX.X.X"
+    return version
+
+
 # Set version string
-try:
-    __version__ = get_distribution(__name__).version
-except DistributionNotFound:
-    # package is not installed
-    __version__ = "vX.X.X"
+__version__ = maybe_get_installed_version(__name__)
 
 
 class Renderer(object):

tests/test_install.py

+# -*- coding:utf-8 -*-
+
+import data_repo_renderer
+import pytest
+
+
+def test_generic_version_if_not_installed():
+    vers = data_repo_renderer.maybe_get_installed_version("datareporenderer")
+    assert vers == "vX.X.X"