| 16 | self.dm.shutdown() |
| 17 | |
| 18 | def testSql(self): |
| 19 | self.dm.sql('set hive.metastore.warehouse.dir=/tmp') |
| 20 | self.dm.sql('drop table if exists airline_na') |
| 21 | self.dm.sql("""create table airline_na (Year int,Month int,DayofMonth int, |
| 22 | DayOfWeek int,DepTime int,CRSDepTime int,ArrTime int, |
| 23 | CRSArrTime int,UniqueCarrier string, FlightNum int, |
| 24 | TailNum string, ActualElapsedTime int, CRSElapsedTime int, |
| 25 | AirTime int, ArrDelay int, DepDelay int, Origin string, |
| 26 | Dest string, Distance int, TaxiIn int, TaxiOut int, Cancelled int, |
| 27 | CancellationCode string, Diverted string, CarrierDelay int, |
| 28 | WeatherDelay int, NASDelay int, SecurityDelay int, LateAircraftDelay int ) |
| 29 | ROW FORMAT DELIMITED FIELDS TERMINATED BY ',' |
| 30 | """) |
| 31 | self.dm.sql("load data local inpath '{}/resources/test/airlineWithNA.csv' " |
| 32 | "into table airline_na".format(DDF_HOME)) |
| 33 | |
| 34 | ddf = self.dm.sql2ddf('select * from airline_na') |
| 35 | self.assertEqual(ddf.rows, 31) |
| 36 | self.assertEqual(ddf.cols, 29) |
| 37 | |
| 38 | if __name__ == '__main__': |
| 39 | unittest.main() |