Skip to content
Snippets Groups Projects

JSON Serializer for pystencils config

Merged Helen Schottenhamml requested to merge mr_json_serializer into master
1 unresolved thread
+ 22
0
 
"""
 
Test the pystencils-specific JSON encoder and serializer as used in the Database class.
 
"""
 
 
import numpy as np
 
import tempfile
 
 
from pystencils.config import CreateKernelConfig
 
from pystencils import Target
 
from pystencils.runhelper.db import Database, PystencilsJsonSerializer
 
 
 
def test_json_serializer():
 
 
# create pystencils config
 
config = CreateKernelConfig(target=Target.CPU, function_name='dummy_config', data_type=np.float32)
 
 
# create dummy database
 
temp_dir = tempfile.TemporaryDirectory()
 
db = Database(file=temp_dir.name, serializer_info=('pystencils_serializer', PystencilsJsonSerializer))
 
 
db.save(params={'config': config}, result={'test': 'dummy'})