Coverage for tests\dataset_test.py: 100%

1from tempfile import TemporaryDirectory 

2 

3from pandas import DataFrame 

4from pyarrow.dataset import dataset 

5 

6from gtrends_collection import write_to_dataset 

7 

8 

9def test_dataset(): 

10 with TemporaryDirectory() as tempdir: 

11 data = DataFrame( 

12 { 

13 "term": ["a", "b"], 

14 "value": [0] * 2, 

15 "date": ["2004-01-01"] * 2, 

16 "location": ["US"] * 2, 

17 "retrieved": ["2025-03-07"] * 2, 

18 } 

19 ) 

20 write_to_dataset(data, tempdir) 

21 assert dataset(tempdir).scanner(["term"]).to_table()["term"].to_pylist() == data["term"].to_list()