@@ -251,19 +251,17 @@ def test_pandas_column_datatype(self):
251251 response_metadata_mode = FluxResponseMetadataMode .full )
252252 df = list (parser .generator ())[0 ]
253253 self .assertEqual (13 , df .dtypes .__len__ ())
254-
255- assert df .dtypes ['result' ].name in ['str' , 'object' ]
256-
254+ self .assertEqual ('str' , df .dtypes ['result' ].name )
257255 self .assertEqual ('int64' , df .dtypes ['table' ].name )
258256 self .assertIn ('datetime64[us,' , df .dtypes ['_start' ].name )
259257 self .assertIn ('datetime64[us,' , df .dtypes ['_stop' ].name )
260- assert df .dtypes ['_field' ].name in [ 'str' , 'object' ]
261- assert df .dtypes ['_measurement' ].name in [ 'str' , 'object' ]
262- assert df .dtypes ['host' ].name in [ 'str' , 'object' ]
263- assert df .dtypes ['region' ].name in [ 'str' , 'object' ]
258+ self . assertEqual ( 'str' , df .dtypes ['_field' ].name )
259+ self . assertEqual ( 'str' , df .dtypes ['_measurement' ].name )
260+ self . assertEqual ( 'str' , df .dtypes ['host' ].name )
261+ self . assertEqual ( 'str' , df .dtypes ['region' ].name )
264262 self .assertEqual ('int64' , df .dtypes ['value1' ].name )
265263 self .assertEqual ('int64' , df .dtypes ['value2' ].name )
266- assert df .dtypes ['value3' ].name in [ 'str' , 'object' ]
264+ self . assertEqual ( 'str' , df .dtypes ['value3' ].name )
267265 self .assertEqual ('bool' , df .dtypes ['value4' ].name )
268266 self .assertEqual ('float64' , df .dtypes ['value5' ].name )
269267
0 commit comments