@@ -251,17 +251,19 @@ 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- self .assertEqual ('str' , df .dtypes ['result' ].name )
254+
255+ assert df .dtypes ['result' ].name in ['str' , 'object' ]
256+
255257 self .assertEqual ('int64' , df .dtypes ['table' ].name )
256258 self .assertIn ('datetime64[us,' , df .dtypes ['_start' ].name )
257259 self .assertIn ('datetime64[us,' , df .dtypes ['_stop' ].name )
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 )
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' ]
262264 self .assertEqual ('int64' , df .dtypes ['value1' ].name )
263265 self .assertEqual ('int64' , df .dtypes ['value2' ].name )
264- self . assertEqual ( 'str' , df .dtypes ['value3' ].name )
266+ assert df .dtypes ['value3' ].name in [ 'str' , 'object' ]
265267 self .assertEqual ('bool' , df .dtypes ['value4' ].name )
266268 self .assertEqual ('float64' , df .dtypes ['value5' ].name )
267269
0 commit comments