@@ -122,16 +122,16 @@ def assert_func(self, e, output, err):
122122 exception_code = e .code
123123 self .assertEqual (exception_code , 0 )
124124 output = output .getvalue ()
125- self .assertTrue ('| name |' in output )
126- self .assertTrue ('| Testing44209 |' in output )
125+ self .assertIn ('| name |' , output )
126+ self .assertIn ('| Testing44209 |' , output )
127127 self ._output_format ('tabular' , assert_func )
128128
129129 def test_json_output (self ):
130130 def assert_func (self , e , output , err ):
131131 exception_code = e .code
132132 self .assertEqual (exception_code , 0 )
133133 output = output .getvalue ()
134- self .assertTrue ('"name": "Testing44209"' in output )
134+ self .assertIn ('"name": "Testing44209"' , output )
135135 self ._output_format ('json' , assert_func )
136136
137137 def test_json_row_output (self ):
@@ -141,7 +141,7 @@ def assert_func(self, e, output, err):
141141 exception_code = e .code
142142 self .assertEqual (exception_code , 0 )
143143 output = output .getvalue ()
144- self .assertTrue ('{"table_name": "cluster"}' in output )
144+ self .assertIn ('{"table_name": "cluster"}' , output )
145145 self ._output_format ('json_row' , assert_func , query )
146146
147147 def test_csv_obj_output (self ):
@@ -151,7 +151,7 @@ def assert_func(self, e, output, err):
151151 exception_code = e .code
152152 self .assertEqual (exception_code , 0 )
153153 output = output .getvalue ()
154- self .assertTrue ("""crate,'{"http": 44209, "psql": 45441, "transport": 44309}'""" in output )
154+ self .assertIn ("""crate,'{"http": 44209, "psql": 45441, "transport": 44309}'""" , output )
155155
156156 self ._output_format ('csv' , assert_func , query )
157157
@@ -162,7 +162,7 @@ def assert_func(self, e, output, err):
162162 exception_code = e .code
163163 self .assertEqual (exception_code , 0 )
164164 output = output .getvalue ()
165- self .assertTrue ('["/dev/' in output )
165+ self .assertIn ('["/dev/' , output )
166166
167167 self ._output_format ('csv' , assert_func , query )
168168
@@ -171,18 +171,18 @@ def assert_func(self, e, output, err):
171171 exception_code = e .code
172172 self .assertEqual (exception_code , 0 )
173173 output = output .getvalue ()
174- self .assertTrue ('"duration":' in output )
175- self .assertTrue ('"rowcount":' in output )
176- self .assertTrue ('"rows":' in output )
177- self .assertTrue ('"cols":' in output )
174+ self .assertIn ('"duration":' , output )
175+ self .assertIn ('"rowcount":' , output )
176+ self .assertIn ('"rows":' , output )
177+ self .assertIn ('"cols":' , output )
178178 self ._output_format ('raw' , assert_func )
179179
180180 def test_mixed_output (self ):
181181 def assert_func (self , e , output , err ):
182182 exception_code = e .code
183183 self .assertEqual (exception_code , 0 )
184184 output = output .getvalue ()
185- self .assertTrue ("name | Testing44209" in output )
185+ self .assertIn ("name | Testing44209" , output )
186186 self ._output_format ('mixed' , assert_func )
187187
188188 def test_pprint_duplicate_keys (self ):
@@ -224,7 +224,7 @@ def test_stdin_cmd(self):
224224 exception_code = e .code
225225 self .assertEqual (exception_code , 0 )
226226 output = output .getvalue ()
227- self .assertTrue ('via-stdin' in output )
227+ self .assertIn ('via-stdin' , output )
228228 finally :
229229 try :
230230 os .remove (tmphistory )
@@ -254,8 +254,8 @@ def test_cmd_precedence(self):
254254 exception_code = e .code
255255 self .assertEqual (exception_code , 0 )
256256 output = output .getvalue ()
257- self .assertTrue ('via-command' in output )
258- self .assertFalse ('via-stdin' in output )
257+ self .assertIn ('via-command' , output )
258+ self .assertNotIn ('via-stdin' , output )
259259 finally :
260260 try :
261261 os .remove (tmphistory )
0 commit comments