artiq/test/serialization.py

39 lines
954 B
Python
Raw Normal View History

2014-12-04 17:52:22 +08:00
import unittest
import json
from fractions import Fraction
import numpy as np
2014-12-09 10:48:47 +08:00
from artiq.language.units import *
2014-12-04 17:52:22 +08:00
from artiq.management import pyon
_pyon_test_object = {
(1, 2): [(3, 4.2), (2, )],
2014-12-09 10:48:47 +08:00
Fraction(3, 4): np.linspace(5, 10, 1),
"frequency": 10*GHz
2014-12-04 17:52:22 +08:00
}
class PYON(unittest.TestCase):
def test_encdec(self):
for enc in pyon.encode, lambda x: pyon.encode(x, True):
self.assertEqual(pyon.decode(enc(_pyon_test_object)),
_pyon_test_object)
_json_test_object = {
"a": "b",
"x": [1, 2, {}],
"foo\nbaz\\qux\"": ["bar", 1.2, {"x": "y"}],
"bar": [True, False, None]
}
class JSONPYON(unittest.TestCase):
def test_encdec(self):
for enc in pyon.encode, lambda x: pyon.encode(x, True), json.dumps:
for dec in pyon.decode, json.loads:
self.assertEqual(dec(enc(_json_test_object)),
_json_test_object)