mirror of
https://github.com/hanzo-ds/driver.git
synced 2026-10-30 20:32:40 +00:00
23 lines
675 B
Python
23 lines
675 B
Python
from tests.numpy.testcase import NumpyBaseTestCase
|
|
|
|
|
|
class ColumnsNamesTestCase(NumpyBaseTestCase):
|
|
|
|
def test_columns_names_replace_nonwords(self):
|
|
columns = (
|
|
'regular Int64, '
|
|
'CamelCase Int64, '
|
|
'With_Underscores Int64, '
|
|
'`Any%different.Column?` Int64'
|
|
)
|
|
|
|
expected_columns = [
|
|
'regular', 'CamelCase', 'With_Underscores', 'Any%different.Column?'
|
|
]
|
|
|
|
with self.create_table(columns):
|
|
df = self.client.query_dataframe(
|
|
'SELECT * FROM test', replace_nonwords=False
|
|
)
|
|
self.assertEqual(expected_columns, list(df.columns))
|