setup.py

from setuptools import setup, find_packages
from imp import load_source
import unittest

pkginfo = load_source("pkginfo.version", "gtirb_stack_stamp/version.py")
__version__ = pkginfo.__version__


def gtirb_stack_stamp_test_suite():
    test_loader = unittest.TestLoader()
    test_suite = test_loader.discover("tests", pattern="test_*.py")
    return test_suite


if __name__ == "__main__":
    setup(
        name="gtirb-stack-stamp",
        version=__version__,
        author="Grammatech",
        author_email="gtirb@grammatech.com",
        description="Apply a stack-stamp transform to GTIRB",
        package_data={"gtirb_stack_stamp": ["gtirb_stack_stamp/*.py"]},
        packages=find_packages(),
        test_suite="setup.gtirb_stack_stamp_test_suite",
        install_requires=["gtirb", "gtirb-rewriting"],
        classifiers=["Programming Language :: Python :: 3"],
        entry_points={
            "console_scripts": [
                "gtirb-stack-stamp = gtirb_stack_stamp.__main__:main"
            ],
            "gtirb_rewriting": [
                "stack-stamp = gtirb_stack_stamp.stack_stamp:StampPass",
            ],
        },
    )