| 204 | self.seq.json(1) |
| 205 | |
| 206 | def test_sqlite3(self): |
| 207 | db_file = "functional/test/data/test_sqlite3.db" |
| 208 | |
| 209 | # test failure case |
| 210 | with self.assertRaises(ValueError): |
| 211 | self.seq.sqlite3(1, "SELECT * from user").to_list() |
| 212 | |
| 213 | # test select from file path |
| 214 | query_0 = "SELECT id, name FROM user;" |
| 215 | result_0 = self.seq.sqlite3(db_file, query_0).to_list() |
| 216 | expected_0 = [(1, "Tom"), (2, "Jack"), (3, "Jane"), (4, "Stephan")] |
| 217 | self.assertListEqual(expected_0, result_0) |
| 218 | |
| 219 | # test select from connection |
| 220 | with sqlite3.connect(db_file) as conn: |
| 221 | result_0_1 = self.seq.sqlite3(conn, query_0).to_list() |
| 222 | self.assertListEqual(expected_0, result_0_1) |
| 223 | |
| 224 | # test select from cursor |
| 225 | with sqlite3.connect(db_file) as conn: |
| 226 | cursor = conn.cursor() |
| 227 | result_0_2 = self.seq.sqlite3(cursor, query_0).to_list() |
| 228 | self.assertListEqual(expected_0, result_0_2) |
| 229 | |
| 230 | # test connection with kwds |
| 231 | result_0_3 = self.seq.sqlite3(db_file, query_0, timeout=30).to_list() |
| 232 | self.assertListEqual(expected_0, result_0_3) |
| 233 | |
| 234 | # test order by |
| 235 | result_1 = self.seq.sqlite3( |
| 236 | db_file, "SELECT id, name FROM user ORDER BY name;" |
| 237 | ).to_list() |
| 238 | expected_1 = [(2, "Jack"), (3, "Jane"), (4, "Stephan"), (1, "Tom")] |
| 239 | self.assertListEqual(expected_1, result_1) |
| 240 | |
| 241 | # test query with params |
| 242 | result_2 = self.seq.sqlite3( |
| 243 | db_file, "SELECT id, name FROM user WHERE id = ?;", parameters=(1,) |
| 244 | ).to_list() |
| 245 | expected_2 = [(1, "Tom")] |
| 246 | self.assertListEqual(expected_2, result_2) |
| 247 | |
| 248 | def test_pandas(self): |
| 249 | try: |