old_contents: #!/usr/bin/env python
  """Test case."""
  
  import os
  import subprocess
  
  try:
      # Python 2.6
      import unittest2 as unittest
  except ImportError:
      import unittest
  
  
  ROOT_DIR = os.path.split(os.path.abspath(os.path.dirname(__file__)))[0]
  
  
  class Tests(unittest.TestCase):
  
      def test_system(self):
  
          subprocess.check_call(
              [os.path.join(ROOT_DIR, 'heatmap.py'),
               '-p', os.path.join(ROOT_DIR, 'test', 'few-points'),
               '-b', 'black',
               '-r', '3',
               '-W', '22',
               '-P', 'equirectangular',
               '-o', os.path.join(ROOT_DIR, 'test', 'output.ppm')])
  
          subprocess.check_call(
              ['perceptualdiff',
               os.path.join(ROOT_DIR, 'test', 'few-points.ppm'),
               os.path.join(ROOT_DIR, 'test', 'output.ppm')])
  
  
  if __name__ == '__main__':
      unittest.main()
  
new_contents: #!/usr/bin/env python
  """Test case."""
  
  import os
  import subprocess
  
  import unittest
  
  
  ROOT_DIR = os.path.split(os.path.abspath(os.path.dirname(__file__)))[0]
  
  
  class Tests(unittest.TestCase):
  
      def test_system(self):
  
          subprocess.check_call(
              [os.path.join(ROOT_DIR, 'heatmap.py'),
               '-p', os.path.join(ROOT_DIR, 'test', 'few-points'),
               '-b', 'black',
               '-r', '3',
               '-W', '22',
               '-P', 'equirectangular',
               '-o', os.path.join(ROOT_DIR, 'test', 'output.ppm')])
  
          subprocess.check_call(
              ['perceptualdiff',
               os.path.join(ROOT_DIR, 'test', 'few-points.ppm'),
               os.path.join(ROOT_DIR, 'test', 'output.ppm')])
  
  
  if __name__ == '__main__':
      unittest.main()
  
subject: Remove unnecessary unittest2 import