old_contents: #!/usr/bin/env python
  
  import unittest
  import os
  import logging
  import sys
  
  # Setup test files and logs
  dir = os.path.dirname(__file__)
  log_file = os.path.join(dir, 'unittest/unit-test.log')
  
  # Logging setup
  logging.basicConfig(filename=log_file, level=logging.DEBUG)
  
  
  class TestPatentConfig(unittest.TestCase):
      # Make sure that if os.environ['PATENTROOT'] is set, then we parse it correctly
      # if it is nonexistant/incorrect, then recover
      # if it is not set, then we default to /data/patentdata/patents
  
      def setUp(self):
          # make sure we can call parse.py using the os module
          # for the purpose of testing command line arguments
          current_directory = os.getcwd()
          if not current_directory.endswith('test'):
              logging.error('Please run from the patentprocessor/test directory')
          # need a default value for later tests
          self.assertTrue(os.environ.has_key('PATENTROOT'))
  
      def test_argparse_patentroot(self):
          # test that argparse is setting the variables correctly for patentroot
          os.chdir('..')
          exit_status = os.system('python parse.py --patentroot %s' % (os.getcwd() + '/unittest/fixtures'))
          # because of the default regex, this should fail
          self.assertTrue(exit_status != 0)
  
          exit_status = os.system('python parse.py --patentroot /dev/null')
          # specify invalid directory, should fail
          self.assertTrue(exit_status != 0)
  
          # test a working, valid directory
          exit_status = os.system('python parse.py --patentroot %s' % (os.environ['PATENTROOT']))
          # this should pass
          self.assertTrue(exit_status == 0)
          # reset directory
          os.chdir('test')
  
  
      def test_gets_environment_var(self):
          # sanity check for existing valid path
          logging.info("Testing getting valid env var PATENTROOT")
          self.assertTrue(os.environ.has_key('PATENTROOT'))
  
  if __name__ == '__main__':
  
      open(log_file, 'w')
      unittest.main()
  
new_contents: #!/usr/bin/env python
  
  import unittest
  import os
  import logging
  import sys
  
  # Setup test files and logs
  dir = os.path.dirname(__file__)
  log_file = os.path.join(dir, 'unittest/unit-test.log')
  
  # Logging setup
  logging.basicConfig(filename=log_file, level=logging.DEBUG)
  
  
  class TestPatentConfig(unittest.TestCase):
      # Make sure that if os.environ['PATENTROOT'] is set, then we parse it correctly
      # if it is nonexistant/incorrect, then recover
      # if it is not set, then we default to /data/patentdata/patents
  
      def setUp(self):
          # make sure we can call parse.py using the os module
          # for the purpose of testing command line arguments
          current_directory = os.getcwd()
          if not current_directory.endswith('test'):
              logging.error('Please run from the patentprocessor/test directory')
          # need a default value for later tests
          self.assertTrue(os.environ.has_key('PATENTROOT'))
  
      def test_argparse_patentroot(self):
          # test that argparse is setting the variables correctly for patentroot
          os.chdir('..')
          exit_status = os.system('python parse.py --patentroot %s' % (os.getcwd() + '/unittest/fixtures'))
          # because of the default regex, this should fail
          self.assertTrue(exit_status != 0)
  
          exit_status = os.system('python parse.py --patentroot /dev/null')
          # specify invalid directory, should fail
          self.assertTrue(exit_status != 0)
  
          # test a working, valid directory
          exit_status = os.system('python parse.py --patentroot %s' % (os.environ['PATENTROOT']))
          # this should pass
          self.assertTrue(exit_status == 0)
          # reset directory
          os.chdir('test')
  
      def test_argparse_directory(self):
          # test that argparse is setting the variables correctly for directories
          os.chdir('..')
          # parse.py should not find any .xml files, so this should fail
          exit_status = os.system('python parse.py --patentroot %s' % (os.getcwd() + '/unittest'))
          self.assertTrue(exit_status != 0)
  
          # parse.py should concatentate the correct directory and find xml files
          exit_status = os.system('python parse.py --patentroot %s --directory fixtures' % (os.getcwd() + '/unittest'))
          self.assertTrue(exit_status == 0)
  
          # TODO: make test for iterating through multiple directories
  
          # reset directory
          os.chdir('test')
  
  
  
      def test_gets_environment_var(self):
          # sanity check for existing valid path
          logging.info("Testing getting valid env var PATENTROOT")
          self.assertTrue(os.environ.has_key('PATENTROOT'))
  
  if __name__ == '__main__':
  
      open(log_file, 'w')
      unittest.main()
  
subject: add test_argparse_directory