@@ -524,12 +524,6 @@ def test_read_sql_view(self):
524
524
"SELECT * FROM iris_view" , self .conn )
525
525
self ._check_iris_loaded_frame (iris_frame )
526
526
527
- def test_legacy_read_frame (self ):
528
- with tm .assert_produces_warning (FutureWarning ):
529
- iris_frame = sql .read_frame (
530
- "SELECT * FROM iris" , self .conn )
531
- self ._check_iris_loaded_frame (iris_frame )
532
-
533
527
def test_to_sql (self ):
534
528
sql .to_sql (self .test_frame1 , 'test_frame1' , self .conn , flavor = 'sqlite' )
535
529
self .assertTrue (
@@ -598,17 +592,6 @@ def test_to_sql_panel(self):
598
592
self .assertRaises (NotImplementedError , sql .to_sql , panel ,
599
593
'test_panel' , self .conn , flavor = 'sqlite' )
600
594
601
- def test_legacy_write_frame (self ):
602
- # Assume that functionality is already tested above so just do
603
- # quick check that it basically works
604
- with tm .assert_produces_warning (FutureWarning ):
605
- sql .write_frame (self .test_frame1 , 'test_frame_legacy' , self .conn ,
606
- flavor = 'sqlite' )
607
-
608
- self .assertTrue (
609
- sql .has_table ('test_frame_legacy' , self .conn , flavor = 'sqlite' ),
610
- 'Table not written to DB' )
611
-
612
595
def test_roundtrip (self ):
613
596
sql .to_sql (self .test_frame1 , 'test_frame_roundtrip' ,
614
597
con = self .conn , flavor = 'sqlite' )
@@ -2239,7 +2222,7 @@ def test_write_row_by_row(self):
2239
2222
2240
2223
self .conn .commit ()
2241
2224
2242
- result = sql .read_frame ("select * from test" , con = self .conn )
2225
+ result = sql .read_sql ("select * from test" , con = self .conn )
2243
2226
result .index = frame .index
2244
2227
tm .assert_frame_equal (result , frame )
2245
2228
@@ -2254,7 +2237,7 @@ def test_execute(self):
2254
2237
sql .execute (ins , self .conn , params = tuple (row ))
2255
2238
self .conn .commit ()
2256
2239
2257
- result = sql .read_frame ("select * from test" , self .conn )
2240
+ result = sql .read_sql ("select * from test" , self .conn )
2258
2241
result .index = frame .index [:1 ]
2259
2242
tm .assert_frame_equal (result , frame [:1 ])
2260
2243
@@ -2327,8 +2310,8 @@ def test_na_roundtrip(self):
2327
2310
pass
2328
2311
2329
2312
def _check_roundtrip (self , frame ):
2330
- sql .write_frame (frame , name = 'test_table' , con = self .conn )
2331
- result = sql .read_frame ("select * from test_table" , self .conn )
2313
+ sql .to_sql (frame , name = 'test_table' , con = self .conn , index = False )
2314
+ result = sql .read_sql ("select * from test_table" , self .conn )
2332
2315
2333
2316
# HACK! Change this once indexes are handled properly.
2334
2317
result .index = frame .index
@@ -2339,8 +2322,8 @@ def _check_roundtrip(self, frame):
2339
2322
frame ['txt' ] = ['a' ] * len (frame )
2340
2323
frame2 = frame .copy ()
2341
2324
frame2 ['Idx' ] = Index (lrange (len (frame2 ))) + 10
2342
- sql .write_frame (frame2 , name = 'test_table2' , con = self .conn )
2343
- result = sql .read_frame ("select * from test_table2" , self .conn ,
2325
+ sql .to_sql (frame2 , name = 'test_table2' , con = self .conn , index = False )
2326
+ result = sql .read_sql ("select * from test_table2" , self .conn ,
2344
2327
index_col = 'Idx' )
2345
2328
expected = frame .copy ()
2346
2329
expected .index = Index (lrange (len (frame2 ))) + 10
@@ -2349,7 +2332,7 @@ def _check_roundtrip(self, frame):
2349
2332
2350
2333
def test_tquery (self ):
2351
2334
frame = tm .makeTimeDataFrame ()
2352
- sql .write_frame (frame , name = 'test_table' , con = self .conn )
2335
+ sql .to_sql (frame , name = 'test_table' , con = self .conn , index = False )
2353
2336
result = sql .tquery ("select A from test_table" , self .conn )
2354
2337
expected = Series (frame .A .values , frame .index ) # not to have name
2355
2338
result = Series (result , frame .index )
@@ -2367,7 +2350,7 @@ def test_tquery(self):
2367
2350
2368
2351
def test_uquery (self ):
2369
2352
frame = tm .makeTimeDataFrame ()
2370
- sql .write_frame (frame , name = 'test_table' , con = self .conn )
2353
+ sql .to_sql (frame , name = 'test_table' , con = self .conn , index = False )
2371
2354
stmt = 'INSERT INTO test_table VALUES(2.314, -123.1, 1.234, 2.3)'
2372
2355
self .assertEqual (sql .uquery (stmt , con = self .conn ), 1 )
2373
2356
@@ -2387,22 +2370,22 @@ def test_keyword_as_column_names(self):
2387
2370
'''
2388
2371
'''
2389
2372
df = DataFrame ({'From' : np .ones (5 )})
2390
- sql .write_frame (df , con = self .conn , name = 'testkeywords' )
2373
+ sql .to_sql (df , con = self .conn , name = 'testkeywords' , index = False )
2391
2374
2392
2375
def test_onecolumn_of_integer (self ):
2393
2376
# GH 3628
2394
2377
# a column_of_integers dataframe should transfer well to sql
2395
2378
2396
2379
mono_df = DataFrame ([1 , 2 ], columns = ['c0' ])
2397
- sql .write_frame (mono_df , con = self .conn , name = 'mono_df' )
2380
+ sql .to_sql (mono_df , con = self .conn , name = 'mono_df' , index = False )
2398
2381
# computing the sum via sql
2399
2382
con_x = self .conn
2400
2383
the_sum = sum ([my_c0 [0 ]
2401
2384
for my_c0 in con_x .execute ("select * from mono_df" )])
2402
2385
# it should not fail, and gives 3 ( Issue #3628 )
2403
2386
self .assertEqual (the_sum , 3 )
2404
2387
2405
- result = sql .read_frame ("select * from mono_df" , con_x )
2388
+ result = sql .read_sql ("select * from mono_df" , con_x )
2406
2389
tm .assert_frame_equal (result , mono_df )
2407
2390
2408
2391
def test_if_exists (self ):
@@ -2421,7 +2404,7 @@ def clean_up(test_table_to_drop):
2421
2404
2422
2405
# test if invalid value for if_exists raises appropriate error
2423
2406
self .assertRaises (ValueError ,
2424
- sql .write_frame ,
2407
+ sql .to_sql ,
2425
2408
frame = df_if_exists_1 ,
2426
2409
con = self .conn ,
2427
2410
name = table_name ,
@@ -2430,34 +2413,34 @@ def clean_up(test_table_to_drop):
2430
2413
clean_up (table_name )
2431
2414
2432
2415
# test if_exists='fail'
2433
- sql .write_frame (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2434
- flavor = 'sqlite' , if_exists = 'fail' )
2416
+ sql .to_sql (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2417
+ flavor = 'sqlite' , if_exists = 'fail' )
2435
2418
self .assertRaises (ValueError ,
2436
- sql .write_frame ,
2419
+ sql .to_sql ,
2437
2420
frame = df_if_exists_1 ,
2438
2421
con = self .conn ,
2439
2422
name = table_name ,
2440
2423
flavor = 'sqlite' ,
2441
2424
if_exists = 'fail' )
2442
2425
2443
2426
# test if_exists='replace'
2444
- sql .write_frame (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2445
- flavor = 'sqlite' , if_exists = 'replace' )
2427
+ sql .to_sql (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2428
+ flavor = 'sqlite' , if_exists = 'replace' , index = False )
2446
2429
self .assertEqual (sql .tquery (sql_select , con = self .conn ),
2447
2430
[(1 , 'A' ), (2 , 'B' )])
2448
- sql .write_frame (frame = df_if_exists_2 , con = self .conn , name = table_name ,
2449
- flavor = 'sqlite' , if_exists = 'replace' )
2431
+ sql .to_sql (frame = df_if_exists_2 , con = self .conn , name = table_name ,
2432
+ flavor = 'sqlite' , if_exists = 'replace' , index = False )
2450
2433
self .assertEqual (sql .tquery (sql_select , con = self .conn ),
2451
2434
[(3 , 'C' ), (4 , 'D' ), (5 , 'E' )])
2452
2435
clean_up (table_name )
2453
2436
2454
2437
# test if_exists='append'
2455
- sql .write_frame (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2456
- flavor = 'sqlite' , if_exists = 'fail' )
2438
+ sql .to_sql (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2439
+ flavor = 'sqlite' , if_exists = 'fail' , index = False )
2457
2440
self .assertEqual (sql .tquery (sql_select , con = self .conn ),
2458
2441
[(1 , 'A' ), (2 , 'B' )])
2459
- sql .write_frame (frame = df_if_exists_2 , con = self .conn , name = table_name ,
2460
- flavor = 'sqlite' , if_exists = 'append' )
2442
+ sql .to_sql (frame = df_if_exists_2 , con = self .conn , name = table_name ,
2443
+ flavor = 'sqlite' , if_exists = 'append' , index = False )
2461
2444
self .assertEqual (sql .tquery (sql_select , con = self .conn ),
2462
2445
[(1 , 'A' ), (2 , 'B' ), (3 , 'C' ), (4 , 'D' ), (5 , 'E' )])
2463
2446
clean_up (table_name )
@@ -2542,7 +2525,7 @@ def test_write_row_by_row(self):
2542
2525
2543
2526
self .conn .commit ()
2544
2527
2545
- result = sql .read_frame ("select * from test" , con = self .conn )
2528
+ result = sql .read_sql ("select * from test" , con = self .conn )
2546
2529
result .index = frame .index
2547
2530
tm .assert_frame_equal (result , frame )
2548
2531
@@ -2577,7 +2560,7 @@ def test_execute(self):
2577
2560
sql .execute (ins , self .conn , params = tuple (row ))
2578
2561
self .conn .commit ()
2579
2562
2580
- result = sql .read_frame ("select * from test" , self .conn )
2563
+ result = sql .read_sql ("select * from test" , self .conn )
2581
2564
result .index = frame .index [:1 ]
2582
2565
tm .assert_frame_equal (result , frame [:1 ])
2583
2566
@@ -2666,9 +2649,9 @@ def _check_roundtrip(self, frame):
2666
2649
with warnings .catch_warnings ():
2667
2650
warnings .filterwarnings ("ignore" , "Unknown table.*" )
2668
2651
cur .execute (drop_sql )
2669
- sql .write_frame (frame , name = 'test_table' ,
2670
- con = self .conn , flavor = 'mysql' )
2671
- result = sql .read_frame ("select * from test_table" , self .conn )
2652
+ sql .to_sql (frame , name = 'test_table' ,
2653
+ con = self .conn , flavor = 'mysql' , index = False )
2654
+ result = sql .read_sql ("select * from test_table" , self .conn )
2672
2655
2673
2656
# HACK! Change this once indexes are handled properly.
2674
2657
result .index = frame .index
@@ -2686,9 +2669,9 @@ def _check_roundtrip(self, frame):
2686
2669
with warnings .catch_warnings ():
2687
2670
warnings .filterwarnings ("ignore" , "Unknown table.*" )
2688
2671
cur .execute (drop_sql )
2689
- sql .write_frame (frame2 , name = 'test_table2' ,
2690
- con = self .conn , flavor = 'mysql' )
2691
- result = sql .read_frame ("select * from test_table2" , self .conn ,
2672
+ sql .to_sql (frame2 , name = 'test_table2' ,
2673
+ con = self .conn , flavor = 'mysql' , index = False )
2674
+ result = sql .read_sql ("select * from test_table2" , self .conn ,
2692
2675
index_col = 'Idx' )
2693
2676
expected = frame .copy ()
2694
2677
@@ -2706,8 +2689,8 @@ def test_tquery(self):
2706
2689
drop_sql = "DROP TABLE IF EXISTS test_table"
2707
2690
cur = self .conn .cursor ()
2708
2691
cur .execute (drop_sql )
2709
- sql .write_frame (frame , name = 'test_table' ,
2710
- con = self .conn , flavor = 'mysql' )
2692
+ sql .to_sql (frame , name = 'test_table' ,
2693
+ con = self .conn , flavor = 'mysql' , index = False )
2711
2694
result = sql .tquery ("select A from test_table" , self .conn )
2712
2695
expected = Series (frame .A .values , frame .index ) # not to have name
2713
2696
result = Series (result , frame .index )
@@ -2732,8 +2715,8 @@ def test_uquery(self):
2732
2715
drop_sql = "DROP TABLE IF EXISTS test_table"
2733
2716
cur = self .conn .cursor ()
2734
2717
cur .execute (drop_sql )
2735
- sql .write_frame (frame , name = 'test_table' ,
2736
- con = self .conn , flavor = 'mysql' )
2718
+ sql .to_sql (frame , name = 'test_table' ,
2719
+ con = self .conn , flavor = 'mysql' , index = False )
2737
2720
stmt = 'INSERT INTO test_table VALUES(2.314, -123.1, 1.234, 2.3)'
2738
2721
self .assertEqual (sql .uquery (stmt , con = self .conn ), 1 )
2739
2722
@@ -2754,8 +2737,8 @@ def test_keyword_as_column_names(self):
2754
2737
'''
2755
2738
_skip_if_no_pymysql ()
2756
2739
df = DataFrame ({'From' : np .ones (5 )})
2757
- sql .write_frame (df , con = self .conn , name = 'testkeywords' ,
2758
- if_exists = 'replace' , flavor = 'mysql' )
2740
+ sql .to_sql (df , con = self .conn , name = 'testkeywords' ,
2741
+ if_exists = 'replace' , flavor = 'mysql' , index = False )
2759
2742
2760
2743
def test_if_exists (self ):
2761
2744
_skip_if_no_pymysql ()
@@ -2774,7 +2757,7 @@ def clean_up(test_table_to_drop):
2774
2757
2775
2758
# test if invalid value for if_exists raises appropriate error
2776
2759
self .assertRaises (ValueError ,
2777
- sql .write_frame ,
2760
+ sql .to_sql ,
2778
2761
frame = df_if_exists_1 ,
2779
2762
con = self .conn ,
2780
2763
name = table_name ,
@@ -2783,34 +2766,34 @@ def clean_up(test_table_to_drop):
2783
2766
clean_up (table_name )
2784
2767
2785
2768
# test if_exists='fail'
2786
- sql .write_frame (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2787
- flavor = 'mysql' , if_exists = 'fail' )
2769
+ sql .to_sql (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2770
+ flavor = 'mysql' , if_exists = 'fail' , index = False )
2788
2771
self .assertRaises (ValueError ,
2789
- sql .write_frame ,
2772
+ sql .to_sql ,
2790
2773
frame = df_if_exists_1 ,
2791
2774
con = self .conn ,
2792
2775
name = table_name ,
2793
2776
flavor = 'mysql' ,
2794
2777
if_exists = 'fail' )
2795
2778
2796
2779
# test if_exists='replace'
2797
- sql .write_frame (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2798
- flavor = 'mysql' , if_exists = 'replace' )
2780
+ sql .to_sql (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2781
+ flavor = 'mysql' , if_exists = 'replace' , index = False )
2799
2782
self .assertEqual (sql .tquery (sql_select , con = self .conn ),
2800
2783
[(1 , 'A' ), (2 , 'B' )])
2801
- sql .write_frame (frame = df_if_exists_2 , con = self .conn , name = table_name ,
2802
- flavor = 'mysql' , if_exists = 'replace' )
2784
+ sql .to_sql (frame = df_if_exists_2 , con = self .conn , name = table_name ,
2785
+ flavor = 'mysql' , if_exists = 'replace' , index = False )
2803
2786
self .assertEqual (sql .tquery (sql_select , con = self .conn ),
2804
2787
[(3 , 'C' ), (4 , 'D' ), (5 , 'E' )])
2805
2788
clean_up (table_name )
2806
2789
2807
2790
# test if_exists='append'
2808
- sql .write_frame (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2809
- flavor = 'mysql' , if_exists = 'fail' )
2791
+ sql .to_sql (frame = df_if_exists_1 , con = self .conn , name = table_name ,
2792
+ flavor = 'mysql' , if_exists = 'fail' , index = False )
2810
2793
self .assertEqual (sql .tquery (sql_select , con = self .conn ),
2811
2794
[(1 , 'A' ), (2 , 'B' )])
2812
- sql .write_frame (frame = df_if_exists_2 , con = self .conn , name = table_name ,
2813
- flavor = 'mysql' , if_exists = 'append' )
2795
+ sql .to_sql (frame = df_if_exists_2 , con = self .conn , name = table_name ,
2796
+ flavor = 'mysql' , if_exists = 'append' , index = False )
2814
2797
self .assertEqual (sql .tquery (sql_select , con = self .conn ),
2815
2798
[(1 , 'A' ), (2 , 'B' ), (3 , 'C' ), (4 , 'D' ), (5 , 'E' )])
2816
2799
clean_up (table_name )
0 commit comments