Esempio n. 1
0
 def test_simple_case(self):
     lists = [[0], [1.125, 1.25]]
     packed, offset = util.pack_arrays(lists)
     assert list(offset) == [0, 1, 3]
     assert list(packed) == [0, 1.125, 1.25]
     returned = util.unpack_arrays(packed, offset)
     for a1, a2 in itertools.zip_longest(lists, returned):
         assert a1 == list(a2)
Esempio n. 2
0
 def test_simple_case(self):
     lists = [[0], [1.125, 1.25]]
     packed, offset = util.pack_arrays(lists)
     self.assertEqual(list(offset), [0, 1, 3])
     self.assertEqual(list(packed), [0, 1.125, 1.25])
     returned = util.unpack_arrays(packed, offset)
     for a1, a2 in itertools.zip_longest(lists, returned):
         self.assertEqual(a1, list(a2))
Esempio n. 3
0
 def verify_packing(self, data):
     packed, offset = util.pack_arrays(data)
     assert packed.dtype == np.float64
     assert offset.dtype == np.uint32
     assert packed.shape[0] == offset[-1]
     returned = util.unpack_arrays(packed, offset)
     for a1, a2 in itertools.zip_longest(data, returned):
         assert np.array_equal(a1, a2)
     return returned
Esempio n. 4
0
 def verify_packing(self, data):
     packed, offset = util.pack_arrays(data)
     self.assertEqual(packed.dtype, np.float64)
     self.assertEqual(offset.dtype, np.uint32)
     self.assertEqual(packed.shape[0], offset[-1])
     returned = util.unpack_arrays(packed, offset)
     for a1, a2 in itertools.zip_longest(data, returned):
         self.assertTrue(np.array_equal(a1, a2))
     return returned