Update DEVELOPERS.md
[datasette-pytables.git] / tests / fixtures.py
1 from datasette.app import Datasette
2 import numpy as np
3 import os
4 from tables import *
5 import tempfile
6
7 def app_client(max_returned_rows=None):
8     with tempfile.TemporaryDirectory() as tmpdir:
9         filepath = os.path.join(tmpdir, 'test_tables.h5')
10         populate_file(filepath)
11         ds = Datasette(
12             [filepath],
13             config={
14                 'default_page_size': 50,
15                 'max_returned_rows': max_returned_rows or 1000,
16             }
17         )
18         client = ds.app().test_client
19         client.ds = ds
20         yield client
21
22 def populate_file(filepath):
23     class Particle(IsDescription):
24         identity = StringCol(itemsize=22, dflt=' ', pos=0)
25         idnumber = Int16Col(dflt=1, pos=1)
26         speed = Float32Col(dflt=1, pos=2)
27
28     h5file = open_file(filepath, mode='w')
29     root = h5file.root
30
31     group1 = h5file.create_group(root, 'group1')
32     group2 = h5file.create_group(root, 'group2')
33
34     array1 = h5file.create_array(root, 'array1', ['string', 'array'])
35
36     table1 = h5file.create_table(group1, 'table1', Particle)
37     table2 = h5file.create_table(group2, 'table2', Particle)
38
39     array2 = h5file.create_array(group1, 'array2', [x for x in range(10000)])
40
41     multiarray = h5file.create_array(group2, 'multi', np.arange(1000).reshape(10, 50, 2))
42
43     for table in (table1, table2):
44         row = table.row
45
46         for i in range(10000):
47             row['identity'] = 'This is particle: %2d' % (i)
48             row['idnumber'] = i
49             row['speed'] = i * 2.
50             row.append()
51
52         table.flush()
53
54     h5file.close()