check4.py

import bison
import numpy
import os

class B:
    def __init__(self,n):
        self.data = numpy.arange(n).astype('i4')
        
class A:
    def __init__(self,n):
        self.n = n
        self.name = 'a class'
        self.b = B(self.n)
        self.data0 = numpy.random.rand(4)
        
    def __call__(self):
        return numpy.sum(self.b.data)
    
aclass = A(45)
array = numpy.random.rand(128)

f = bison.FILE('test4', mode='w')
f.write('aclass', aclass)
f.write('array', array)

ff = bison.FILE('test4', mode='r')
res1 = ff.read()
print('Reading a class without decoder', res1)


class Adec:
    def __init__(self):
        self.type = '__main__.A'
        
    def decode(self, obj):
        out = A(obj['n'])
        return out

class Bdec:
    def __init__(self):
        self.type = '__main__.B'
        
    def decode(self, obj):
        out = B(len(obj['data']))
        out.data = obj['data']
        return out

res2 = ff.read(decoder = [Adec, Bdec])

assert (res2['aclass'].n == aclass.n)
assert (res2['aclass'].name == aclass.name)
assert numpy.all(res2['aclass'].b.data == aclass.b.data)

assert numpy.all(res2['array'] == array)

f.close()
os.popen('rm test4')