@@ -1738,7 +1738,8 @@ def test_illegal_names(self):
1738
1738
1739
1739
for ndx , weird_name in enumerate (['test_weird_name]' ,'test_weird_name[' ,
1740
1740
'test_weird_name`' ,'test_weird_name"' , 'test_weird_name\' ' ,
1741
- '_b.test_weird_name_01-30' , '"_b.test_weird_name_01-30"' ]):
1741
+ '_b.test_weird_name_01-30' , '"_b.test_weird_name_01-30"' ,
1742
+ '12345' ,'12345blah' ]):
1742
1743
df .to_sql (weird_name , self .conn , flavor = self .flavor )
1743
1744
sql .table_exists (weird_name , self .conn )
1744
1745
@@ -1839,16 +1840,30 @@ def test_to_sql_save_index(self):
1839
1840
self ._to_sql_save_index ()
1840
1841
1841
1842
def test_illegal_names (self ):
1843
+ df = DataFrame ([[1 , 2 ], [3 , 4 ]], columns = ['a' , 'b' ])
1844
+
1845
+ # These tables and columns should be ok
1846
+ for ndx , ok_name in enumerate (['99beginswithnumber' ,'12345' ]):
1847
+ df .to_sql (ok_name , self .conn , flavor = self .flavor , index = False ,
1848
+ if_exists = 'replace' )
1849
+ self .conn .cursor ().execute ("DROP TABLE `%s`" % ok_name )
1850
+ self .conn .commit ()
1851
+ df2 = DataFrame ([[1 , 2 ], [3 , 4 ]], columns = ['a' , ok_name ])
1852
+ c_tbl = 'test_ok_col_name%d' % ndx
1853
+ df2 .to_sql (c_tbl , self .conn , flavor = self .flavor , index = False ,
1854
+ if_exists = 'replace' )
1855
+ self .conn .cursor ().execute ("DROP TABLE `%s`" % c_tbl )
1856
+ self .conn .commit ()
1857
+
1842
1858
# For MySQL, these should raise ValueError
1843
1859
for ndx , illegal_name in enumerate (['test_illegal_name]' ,'test_illegal_name[' ,
1844
1860
'test_illegal_name`' ,'test_illegal_name"' , 'test_illegal_name\' ' , '' ]):
1845
- df = DataFrame ([[1 , 2 ], [3 , 4 ]], columns = ['a' , 'b' ])
1846
1861
self .assertRaises (ValueError , df .to_sql , illegal_name , self .conn ,
1847
1862
flavor = self .flavor , index = False )
1848
1863
1849
1864
df2 = DataFrame ([[1 , 2 ], [3 , 4 ]], columns = ['a' , illegal_name ])
1850
1865
c_tbl = 'test_illegal_col_name%d' % ndx
1851
- self .assertRaises (ValueError , df2 .to_sql , 'test_illegal_col_name' ,
1866
+ self .assertRaises (ValueError , df2 .to_sql , c_tbl ,
1852
1867
self .conn , flavor = self .flavor , index = False )
1853
1868
1854
1869
0 commit comments