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
36
37
|
from hdmf.data_utils import append_data
from hdmf.testing import TestCase
import numpy as np
from numpy.testing import assert_array_equal
try:
import zarr
ZARR_INSTALLED = True
except ImportError:
ZARR_INSTALLED = False
class MyIterable:
def __init__(self, data):
self.data = data
class TestAppendData(TestCase):
def test_append_exception(self):
data = MyIterable([1, 2, 3, 4, 5])
with self.assertRaises(ValueError):
append_data(data, 4)
class TestZarrAppendData(TestCase):
def setUp(self):
if not ZARR_INSTALLED:
self.skipTest("optional Zarr package is not installed")
def test_append_data_zarr(self):
zarr_array = zarr.array([1,2,3])
new = append_data(zarr_array, 4)
assert_array_equal(new[:], np.array([1,2,3,4]))
|
