1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
# fmt: off
import pytest
import subprocess
import sys
from typing import List
from conftest import ShellTest
import os
from pathlib import Path
# test import from a parquet file
def test_import_parquet(shell):
test = (
ShellTest(shell)
.statement(f'.import data/parquet-testing/unsigned.parquet a')
.statement("SELECT MAX(d) FROM a")
)
result = test.run()
result.check_stdout("18446744073709551615")
# test import from a json file
def test_import_json(shell):
test = (
ShellTest(shell)
.statement(f'.import data/json/example_n.ndjson a')
.statement("SELECT name FROM a")
)
result = test.run()
result.check_stdout("Broadcast News")
# fmt: on
|
