def test_multiple_csvs_to_sqlite3(self): csv1 = open("data/csv_sqlite_test.csv", "r") csv2 = open("data/second_test.csv", "r") header = [["hi", "int"], ["low", "int"], ["whatever", "text"]] csv1.readline() csv2.readline() files = [csv1,csv2] self.clear_sqlite3_tables() c, conn = seamstress.csv_to_sqlite3(files, "tester", header) c.execute("""SELECT * from tester;""") results = c.fetchall()
def test_multiple_tables(self): csv1 = open("data/csv_sqlite_test.csv", "r") header1 = [["hi", "int"], ["low", "int"], ["whatever", "text"]] csv1.readline() self.clear_sqlite3_tables() c, conn = seamstress.csv_to_sqlite3(csv1, "tester", header1, db="temp/output.sqlite3") csv2 = open("data/different_table.csv", "r") header2 = csv2.readline().split(",") header2 = [h.strip() for h in header2] header2 = [[h, "int"] for h in header2] c, conn = seamstress.csv_to_sqlite3(csv2, "alt", header2, db="temp/output.sqlite3") c.execute("""SELECT * from tester;""") d1 = c.fetchall() print d1 c.execute("""SELECT * from alt;""") d2 = c.fetchall() print d2
def test_csv_to_sqlite3(self): csvf = open("data/csv_sqlite_test.csv", "r") header = [["hi", "int"], ["low", "int"], ["whatever", "text"]] csvf.readline() self.clear_sqlite3_tables() c, conn = seamstress.csv_to_sqlite3(csvf, "tester", header) c.execute("""SELECT * from tester;""") results = c.fetchall() csvf = open("data/csv_sqlite_test.csv", "r") re = csv.reader(csvf) actual_results = [] re.next() for line in re: actual_results.append(line)