old_contents: from distutils.core import setup, Command
  
  
  def discover_and_run_tests():
      import os
      import sys
      import unittest
  
      # get setup.py directory
      setup_file = sys.modules['__main__'].__file__
      setup_dir = os.path.abspath(os.path.dirname(setup_file))
  
      # use the default shared TestLoader instance
      test_loader = unittest.defaultTestLoader
  
      # use the basic test runner that outputs to sys.stderr
      test_runner = unittest.TextTestRunner()
  
      # automatically discover all tests
      # NOTE: only works for python 2.7 and later
      test_suite = test_loader.discover(setup_dir)
  
      # run the test suite
      result = test_runner.run(test_suite)
      if len(result.failures) + len(result.errors) > 0:
          exit(1)
  
  
  class DiscoverTest(Command):
      user_options = []
  
      def initialize_options(self):
          pass
  
      def finalize_options(self):
          pass
  
      def run(self):
          discover_and_run_tests()
  
  setup(
      name='isc_dhcp_leases',
      version='0.7.0',
      packages=['isc_dhcp_leases'],
      url='https://github.com/MartijnBraam/python-isc-dhcp-leases',
      license='MIT',
      author='Martijn Braam',
      author_email='martijn@brixit.nl',
      description='Small python module for reading /var/lib/dhcp/dhcpd.leases from isc-dhcp-server',
      cmdclass={'test': DiscoverTest},
  )
  
new_contents: from distutils.core import setup, Command
  
  
  def discover_and_run_tests():
      import os
      import sys
      import unittest
  
      # get setup.py directory
      setup_file = sys.modules['__main__'].__file__
      setup_dir = os.path.abspath(os.path.dirname(setup_file))
  
      # use the default shared TestLoader instance
      test_loader = unittest.defaultTestLoader
  
      # use the basic test runner that outputs to sys.stderr
      test_runner = unittest.TextTestRunner()
  
      # automatically discover all tests
      # NOTE: only works for python 2.7 and later
      test_suite = test_loader.discover(setup_dir)
  
      # run the test suite
      result = test_runner.run(test_suite)
      if len(result.failures) + len(result.errors) > 0:
          exit(1)
  
  
  class DiscoverTest(Command):
      user_options = []
  
      def initialize_options(self):
          pass
  
      def finalize_options(self):
          pass
  
      def run(self):
          discover_and_run_tests()
  
  setup(
      name='isc_dhcp_leases',
      version='0.7.0',
      packages=['isc_dhcp_leases'],
      url='https://github.com/MartijnBraam/python-isc-dhcp-leases',
      install_requires=['six'],
      license='MIT',
      author='Martijn Braam',
      author_email='martijn@brixit.nl',
      description='Small python module for reading /var/lib/dhcp/dhcpd.leases from isc-dhcp-server',
      cmdclass={'test': DiscoverTest},
  )
  
subject: add six as requirement