Coverage for tests/test_colarray.py: 100%

1import numpy as np 

2import gamdpy as gp 

3 

4def test_colarray(): 

5 size = (100,2) 

6 column_names = ('r', 'v', 'f') 

7 

8 # setup a column array, and save it 

9 ca = gp.colarray(column_names=column_names, size=size) 

10 for column_name in column_names: 

11 ca[column_name] = np.random.uniform(size=size) 

12 ca.save('my_colarray') 

13 

14 # load into new object, and check consistency 

15 ca2 = gp.colarray.load('my_colarray') 

16 for col in ca2.column_names: 

17 assert np.all(ca2[col]==ca[col]) 

18 

19 # check copy'ing 

20 ca3 = ca.copy() 

21 for col in ca3.column_names: 

22 assert np.all(ca3[col]==ca[col]) 

23 

24 # clean up 

25 gp.colarray.remove_files('my_colarray') 

26 

27if __name__ == "__main__": # pragma: no cover 

28 test_colarray()