2525 from StringIO import StringIO
2626
2727try :
28- import xml .etree .cElementTree as ET
28+ import xml .etree .cElementTree as ET
2929except ImportError :
30- import xml .etree .ElementTree as ET
30+ import xml .etree .ElementTree as ET
3131
3232class ScriptTest (unittest .TestCase ):
3333 def test_error_on_script_with_null_scheme (self ):
@@ -45,17 +45,17 @@ def stream_events(self, inputs, ew):
4545
4646 script = NewScript ()
4747
48- _out = StringIO ()
49- _err = StringIO ()
50- ew = EventWriter (_out , _err )
48+ out = StringIO ()
49+ err = StringIO ()
50+ ew = EventWriter (out , err )
5151
52- _in = StringIO ()
52+ in_stream = StringIO ()
5353
5454 args = ["--scheme" ]
55- return_value = script .run_script (args , ew , _in )
55+ return_value = script .run_script (args , ew , in_stream )
5656
57- self .assertEqual ("" , _out .getvalue ())
58- self .assertEqual ("FATAL Modular input script returned a null scheme.\n " , _err .getvalue ())
57+ self .assertEqual ("" , out .getvalue ())
58+ self .assertEqual ("FATAL Modular input script returned a null scheme.\n " , err .getvalue ())
5959 self .assertNotEqual (0 , return_value )
6060
6161 def test_scheme_properly_generated_by_script (self ):
@@ -89,17 +89,17 @@ def stream_events(self, inputs, ew):
8989
9090 script = NewScript ()
9191
92- _out = StringIO ()
93- _err = StringIO ()
94- ew = EventWriter (_out , _err )
92+ out = StringIO ()
93+ err = StringIO ()
94+ ew = EventWriter (out , err )
9595
9696 args = ["--scheme" ]
97- return_value = script .run_script (args , ew , _err )
97+ return_value = script .run_script (args , ew , err )
9898
99- self .assertEqual ("" , _err .getvalue ())
99+ self .assertEqual ("" , err .getvalue ())
100100 self .assertEqual (0 , return_value )
101101
102- found = ET .fromstring (_out .getvalue ())
102+ found = ET .fromstring (out .getvalue ())
103103 expected = ET .parse (open ("data/scheme_without_defaults.xml" )).getroot ()
104104
105105 self .assertTrue (xml_compare (expected , found ))
@@ -122,16 +122,16 @@ def stream_events(self, inputs, ew):
122122
123123 script = NewScript ()
124124
125- _out = StringIO ()
126- _err = StringIO ()
127- ew = EventWriter (_out , _err )
125+ out = StringIO ()
126+ err = StringIO ()
127+ ew = EventWriter (out , err )
128128
129129 args = ["--validate-arguments" ]
130130
131131 return_value = script .run_script (args , ew , open ("data/validation.xml" ))
132132
133- self .assertEqual ("" , _err .getvalue ())
134- self .assertEqual ("" , _out .getvalue ())
133+ self .assertEqual ("" , err .getvalue ())
134+ self .assertEqual ("" , out .getvalue ())
135135 self .assertEqual (0 , return_value )
136136
137137 def test_failed_validation (self ):
@@ -151,18 +151,18 @@ def stream_events(self, inputs, ew):
151151
152152 script = NewScript ()
153153
154- _out = StringIO ()
155- _err = StringIO ()
156- ew = EventWriter (_out , _err )
154+ out = StringIO ()
155+ err = StringIO ()
156+ ew = EventWriter (out , err )
157157
158158 args = ["--validate-arguments" ]
159159
160160 return_value = script .run_script (args , ew , open ("data/validation.xml" ))
161161
162162 expected = ET .parse (open ("data/validation_error.xml" )).getroot ()
163- found = ET .fromstring (_out .getvalue ())
163+ found = ET .fromstring (out .getvalue ())
164164
165- self .assertEqual ("" , _err .getvalue ())
165+ self .assertEqual ("" , err .getvalue ())
166166 self .assertTrue (xml_compare (expected , found ))
167167 self .assertNotEqual (0 , return_value )
168168
@@ -193,17 +193,17 @@ def stream_events(self, inputs, ew):
193193 script = NewScript ()
194194 input_configuration = open ("data/conf_with_2_inputs.xml" )
195195
196- _out = StringIO ()
197- _err = StringIO ()
198- ew = EventWriter (_out , _err )
196+ out = StringIO ()
197+ err = StringIO ()
198+ ew = EventWriter (out , err )
199199
200200 return_value = script .run_script ([], ew , input_configuration )
201201
202202 self .assertEqual (0 , return_value )
203- self .assertEqual ("" , _err .getvalue ())
203+ self .assertEqual ("" , err .getvalue ())
204204
205205 expected = ET .parse (open ("data/stream_with_two_events.xml" )).getroot ()
206- found = ET .fromstring (_out .getvalue ())
206+ found = ET .fromstring (out .getvalue ())
207207
208208 self .assertTrue (xml_compare (expected , found ))
209209
0 commit comments