コード例 #1
0
ファイル: tests.py プロジェクト: hamilton/Seamstress
	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()
コード例 #2
0
ファイル: tests.py プロジェクト: hamilton/Seamstress
	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
コード例 #3
0
ファイル: tests.py プロジェクト: hamilton/Seamstress
	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)