setup.py

import os
import re
import sys
import sysconfig
import platform
import subprocess

from distutils.version import LooseVersion
from setuptools import setup, Extension
from setuptools.command.build_ext import build_ext
from setuptools.command.test import test as TestCommand

class PyTest(TestCommand):
    def run_tests(self):
        import shlex
        #import here, cause outside the eggs aren't loaded
        import pytest
        errno = pytest.main(['tests/'])
        sys.exit(errno)

setup(
    name='tephigram',
    version='0.2.0',
    author='Leif Denby',
    author_email='l.c.denby@leeds.ac.uk',
    description='Tephigram plotting and profile integration',
    long_description='',
    # add custom build_ext command
    cmdclass=dict(test=PyTest),
    zip_safe=False,
    tests_require=['pytest'],
    install_requires=['attrdict',],
    packages=["tephigram"],
)