Skip to content

Commit 594dac0

Browse files
fix: test_pandas_column_datatype data type
1 parent a869b1a commit 594dac0

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

tests/test_FluxCSVParser.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)