Moved spmd tests into separate classes.

1 files changed, 29 insertions, 19 deletions

diff --git a/tests/spmd_test.py b/tests/spmd_test.py
index ab371cb..bbd03b6 100644
--- a/tests/spmd_test.py
+++ b/tests/spmd_test.py
@@ -11,8 +11,7 @@ from screenplain.types import (
 from screenplain.richstring import plain, italic, empty_string
 
 
-class ParseTests(unittest2.TestCase):
-
+class SlugTests(unittest2.TestCase):
     def test_slug_with_prefix(self):
         paras = list(parse([
             'INT. SOMEWHERE - DAY',
@@ -78,6 +77,8 @@ class ParseTests(unittest2.TestCase):
             paras[0].line
         )
 
+
+class SectionTests(unittest2.TestCase):
     def test_section_parsed_correctly(self):
         paras = parse([
             '# first level',
@@ -90,6 +91,8 @@ class ParseTests(unittest2.TestCase):
         self.assertEquals(2, paras[1].level)
         self.assertEquals(plain('second level'), paras[1].text)
 
+
+class DialogTests(unittest2.TestCase):
     # A Character element is any line entirely in caps, with one empty
     # line before it and without an empty line after it.
     def test_all_caps_is_character(self):
@@ -187,8 +190,26 @@ class ParseTests(unittest2.TestCase):
             (False, plain('Nice retirement.'))
         ], dialog.blocks)
 
-    def test_standard_transition(self):
+    def test_leading_and_trailing_spaces_in_dialog(self):
+        paras = list(parse([
+            'JULIET',
+            'O Romeo, Romeo! wherefore art thou Romeo?',
+            '  Deny thy father and refuse thy name;  ',
+            'Or, if thou wilt not, be but sworn my love,',
+            " And I'll no longer be a Capulet.",
+        ]))
+        self.assertEquals([Dialog], [type(p) for p in paras])
+        self.assertEquals([
+            (False, plain(u'O Romeo, Romeo! wherefore art thou Romeo?')),
+            (False, plain(u'Deny thy father and refuse thy name;')),
+            (False, plain(u'Or, if thou wilt not, be but sworn my love,')),
+            (False, plain(u"And I'll no longer be a Capulet.")),
+        ], paras[0].blocks)
+
+
+class TransitionTests(unittest2.TestCase):
 
+    def test_standard_transition(self):
         paras = list(parse([
             'Jack begins to argue vociferously in Vietnamese (?)',
             '',
@@ -262,6 +283,9 @@ class ParseTests(unittest2.TestCase):
         self.assertEquals([Action, Transition], [type(p) for p in paras])
         self.assertEquals(plain('FADE OUT.'), paras[1].line)
 
+
+class ActionTests(unittest2.TestCase):
+
     def test_action_preserves_leading_whitespace(self):
         paras = list(parse([
             'hello',
@@ -277,22 +301,6 @@ class ParseTests(unittest2.TestCase):
             ], paras[1].lines
         )
 
-    def test_leading_and_trailing_spaces_in_dialog(self):
-        paras = list(parse([
-            'JULIET',
-            'O Romeo, Romeo! wherefore art thou Romeo?',
-            '  Deny thy father and refuse thy name;  ',
-            'Or, if thou wilt not, be but sworn my love,',
-            " And I'll no longer be a Capulet.",
-        ]))
-        self.assertEquals([Dialog], [type(p) for p in paras])
-        self.assertEquals([
-            (False, plain(u'O Romeo, Romeo! wherefore art thou Romeo?')),
-            (False, plain(u'Deny thy father and refuse thy name;')),
-            (False, plain(u'Or, if thou wilt not, be but sworn my love,')),
-            (False, plain(u"And I'll no longer be a Capulet.")),
-        ], paras[0].blocks)
-
     def test_single_centered_line(self):
         paras = list(parse(['> center me! <']))
         self.assertEquals([Action], [type(p) for p in paras])
@@ -333,6 +341,8 @@ class ParseTests(unittest2.TestCase):
         self.assertFalse(paras[0].centered)
         self.assertEquals([plain(line) for line in lines], paras[0].lines)
 
+
+class SynopsisTests(unittest2.TestCase):
     def test_synopsis_after_slug_adds_synopsis_to_scene(self):
         paras = list(parse([
             "EXT. BRICK'S PATIO - DAY",