示例#1
0
    def test_write_simple(self, properties):
        """write/read a file with a single item of 30 frames"""
        nframes = 30
        f = np.random.randn(nframes, self.dim)
        t = np.linspace(0, 2, nframes)
        if properties:
            props = {'a': 0, 'b': 'b'}
        else:
            props = None

        h5f.simple_write(
            self.filename, 'group1', t, f,
            properties=props, item='item', mode='w')

        if properties:
            tr, fr, pr = h5f.read(self.filename, 'group1')
            assert list(pr.keys()) == ['item']
            assert pr['item'] == props
        else:
            tr, fr = h5f.read(self.filename, 'group1')

        assert list(tr.keys()) == ['item']
        assert list(fr.keys()) == ['item']
        assert len(tr['item']) == 30
        assert len(fr['item']) == 30
        # assert tr['item'] == t
        assert (fr['item'] == f).all()
示例#2
0
    def test_write_simple(self):
        # write/read a file with a single item
        features_0 = np.random.randn(300, self.dim)
        times_0 = np.linspace(0, 2, 300)
        h5f.simple_write(self.filename, 'group1', times_0, features_0, 'item')
        t0, f0 = h5f.read(self.filename, 'group1')

        times_0_r, features_0_r = h5f.read(self.filename, 'group1')
        assert list(times_0_r.keys()) == ['item']
        assert list(features_0_r.keys ()) == ['item']
        assert all(times_0_r['item'] == times_0)
        assert (features_0_r['item'] == features_0).all()
示例#3
0
 def test_write_simple(self):
     """write/read a file with a single item of 30 frames"""
     nframes = 30
     f = np.random.randn(nframes, self.dim)
     t = np.linspace(0, 2, nframes)
     h5f.simple_write(self.filename, 'group1', t, f, 'item')
     tr, fr = h5f.read(self.filename, 'group1')
     assert list(tr.keys()) == ['item']
     assert list(fr.keys()) == ['item']
     assert len(tr['item']) == 30
     assert len(fr['item']) == 30
     # assert tr['item'] == t
     assert (fr['item'] == f).all()
示例#4
0
 def test_write_simple(self):
     """write/read a file with a single item of 30 frames"""
     nframes = 30
     f = np.random.randn(nframes, self.dim)
     t = np.linspace(0, 2, nframes)
     h5f.simple_write(self.filename, 'group1', t, f, 'item')
     tr, fr = h5f.read(self.filename, 'group1')
     assert list(tr.keys()) == ['item']
     assert list(fr.keys ()) == ['item']
     assert len(tr['item']) == 30
     assert len(fr['item']) == 30
     #assert tr['item'] == t
     assert (fr['item'] == f).all()
示例#5
0
    def test_simple_write(self):
        self.features_0 = np.random.randn(30, 20)
        self.times_0 = np.linspace(0, 2, 30)

        h5f.simple_write(self.filename, 'f', self.times_0, self.features_0)

        with h5py.File(self.filename, 'r') as f:
            assert ['f'] == list(f.keys())

            g = f.get('f')
            assert list(g.keys()) == (['features', 'index', 'items', 'labels'])

            assert g['features'].shape == (30, 20)
            assert g['items'].shape == (1, )
            assert g['labels'].shape == (30, )
            assert g['index'].shape == (1, )
示例#6
0
    def test_simple_write(self):
        self.features_0 = np.random.randn(30, 20)
        self.times_0 = np.linspace(0, 2, 30)

        h5f.simple_write(
            self.filename, 'f', self.times_0, self.features_0)

        with h5py.File(self.filename, 'r') as f:
            assert ['f'] == list(f.keys())

            g = f.get('f')
            assert list(g.keys()) == (
                ['features', 'index', 'items', 'labels'])

            assert g['features'].shape == (30, 20)
            assert g['items'].shape == (1,)
            assert g['labels'].shape == (30,)
            assert g['index'].shape == (1,)