Added further tests, PEP8 cleanup

tests/testtvorganise.py

@@ -23,13 +23,20 @@ class testTvOrganise(unittest.TestCase):
 
     def testConfigSettings(self):
         """
-        Using a predefined dict, save then load the config and validate 
+        Using a predefined dict, save then load the config and validate
         the contents
         """
         pass
 
+    def testParseFilenames(self):
+        files = ['/media/out/Heroes - [01x05] - Llamaggeddon.avi']
+
+        res = self.tvo.parse_filenames(files)
+        self.assertEqual(res[0]['showname'], 'Heroes')
+
     def setUp(self):
         self.tvo = tvorganise.TvOrganiser()
+        self.tvo._config = tvorganise.config.Config('tvorganise.cfg')
 
 
 class testFindFiles(unittest.TestCase):
@@ -41,14 +48,14 @@ class testFindFiles(unittest.TestCase):
         os.makedirs("/tmp/find-files-test/folder1")
         os.makedirs("/tmp/find-files-test/folder2")
 
-	open('/tmp/find-files-test/folder1/file1', 'w').close() 
+	open('/tmp/find-files-test/folder1/file1', 'w').close()
         open('/tmp/find-files-test/folder1/file2', 'w').close()
         open('/tmp/find-files-test/folder2/file3', 'w').close()
         open('/tmp/find-files-test/folder2/file4', 'w').close()
         open('/tmp/find-files-test/folder2/file5', 'w').close()
 
     def tearDown(self):
-        shutil.rmtree("/tmp/find-files-test/")        
+        shutil.rmtree("/tmp/find-files-test/")
 
     def testFolderList(self):
         self.assertEqual(len(tvorganise.find_files("/tmp/find-files-test/")),5)
@@ -69,7 +76,7 @@ def suite(suite=None):
 
     suite.addTest(unittest.makeSuite(testTvOrganise))
     suite.addTest(unittest.makeSuite(testFindFiles))
- 
+
     return suite
 
 def run():

tvorganise/__init__.py

@@ -13,11 +13,10 @@ import sys
 import re
 from optparse import OptionParser
 import shutil
-import ConfigParser
 import logging
-
 import config
 
+
 def same_partition(path1, path2):
     """
     Checks to see if two paths are on the same device, returns a

tvorganise/config.py

@@ -1,20 +1,21 @@
 import ConfigParser
 import re
 
+
 def defaults():
-   """
-   Creates a ConfigParser instance and fills it with the default settings
-   """
+    """
+    Creates a ConfigParser instance and fills it with the default settings
+    """
 
-   config = ConfigParser.RawConfigParser()
+    config = ConfigParser.RawConfigParser()
 
-   config.add_section('main')
-   config.set('main', 'target_path', '/media/%(showname)s/Season %(seasonnum)s/')
+    config.add_section('main')
+    config.set('main', 'target_path', '/media/%(showname)s/Season %(seasonnum)s/')
 
-   config.add_section('regex')
-   config.set('regex', 'valid_in_names', "[\\w\\(\\).,\\[\\]'\\ \\-?!#:]")
+    config.add_section('regex')
+    config.set('regex', 'valid_in_names', "[\\w\\(\\).,\\[\\]'\\ \\-?!#:]")
 
-   return config
+    return config
 
 
 class Config(dict):