old_contents: # -*- coding: utf-8 -*-
  """Tests for the source type objects."""
  
  import unittest
  
  from artifacts import errors
  from artifacts import source_type
  
  
  class SourceTypeTest(unittest.TestCase):
    """Class to test the artifact source type."""
  
  
  class ArtifactSourceTypeTest(unittest.TestCase):
    """Class to test the artifacts source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.ArtifactSourceType(names=[u'test'])
  
  
  class FileSourceTypeTest(unittest.TestCase):
    """Class to test the files source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.FileSourceType(paths=[u'test'])
      source_type.FileSourceType(paths=[u'test'], separator=u'\\')
  
  
  class PathSourceTypeTest(unittest.TestCase):
    """Class to test the paths source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.PathSourceType(paths=[u'test'])
      source_type.PathSourceType(paths=[u'test'], separator=u'\\')
  
  
  class WindowsRegistryKeySourceTypeTest(unittest.TestCase):
    """Class to test the Windows Registry keys source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.WindowsRegistryKeySourceType(keys=[ur'HKEY_LOCAL_MACHINE\test'])
  
  
  class WindowsRegistryValueSourceTypeTest(unittest.TestCase):
    """Class to test the Windows Registry value source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.WindowsRegistryValueSourceType(
          key_value_pairs=[{'key': ur'HKEY_LOCAL_MACHINE\test', 'value': u'test'}])
  
      with self.assertRaises(errors.FormatError):
        source_type.WindowsRegistryValueSourceType(
            key_value_pairs=[{'bad': u'test', 'value': u'test'}])
  
      with self.assertRaises(errors.FormatError):
        source_type.WindowsRegistryValueSourceType(
            key_value_pairs={'bad': u'test', 'value': u'test'})
  
  
  class WMIQuerySourceType(unittest.TestCase):
    """Class to test the WMI query source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.WMIQuerySourceType(query=u'test')
  
  
  if __name__ == '__main__':
    unittest.main()
  
new_contents: # -*- coding: utf-8 -*-
  """Tests for the source type objects."""
  
  import unittest
  
  from artifacts import errors
  from artifacts import source_type
  
  
  class SourceTypeTest(unittest.TestCase):
    """Class to test the artifact source type."""
  
  
  class ArtifactSourceTypeTest(unittest.TestCase):
    """Class to test the artifacts source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.ArtifactSourceType(names=[u'test'])
  
  
  class FileSourceTypeTest(unittest.TestCase):
    """Class to test the files source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.FileSourceType(paths=[u'test'])
      source_type.FileSourceType(paths=[u'test'], separator=u'\\')
  
  
  class PathSourceTypeTest(unittest.TestCase):
    """Class to test the paths source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.PathSourceType(paths=[u'test'])
      source_type.PathSourceType(paths=[u'test'], separator=u'\\')
  
  
  class WindowsRegistryKeySourceTypeTest(unittest.TestCase):
    """Class to test the Windows Registry keys source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.WindowsRegistryKeySourceType(keys=[u'HKEY_LOCAL_MACHINE\\test'])
  
  
  class WindowsRegistryValueSourceTypeTest(unittest.TestCase):
    """Class to test the Windows Registry value source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.WindowsRegistryValueSourceType(
          key_value_pairs=[{'key': u'HKEY_LOCAL_MACHINE\\test', 'value': u'test'}])
  
      with self.assertRaises(errors.FormatError):
        source_type.WindowsRegistryValueSourceType(
            key_value_pairs=[{'bad': u'test', 'value': u'test'}])
  
      with self.assertRaises(errors.FormatError):
        source_type.WindowsRegistryValueSourceType(
            key_value_pairs={'bad': u'test', 'value': u'test'})
  
  
  class WMIQuerySourceType(unittest.TestCase):
    """Class to test the WMI query source type."""
  
    def testInitialize(self):
      """Tests the __init__ function."""
      source_type.WMIQuerySourceType(query=u'test')
  
  
  if __name__ == '__main__':
    unittest.main()
  
subject: remove raw unicode strings as they aren't supported in py3