Put version setting to a function

import textwrap

import yaml

# Set version string

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(__name__).version

        version = get_distribution(pkg_name).version

    except DistributionNotFound:

        # package is not installed

    __version__ = "vX.X.X"

        version = "vX.X.X"

    return version

# Set version string

__version__ = maybe_get_installed_version(__name__)

class Renderer(object):

# -*- 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"