ipython/IPython/kernel/tests/test_newserialized.py at html-notebook · commandodev/ipython · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
# encoding: utf-8

"""This file contains unittests for the shell.py module."""

__docformat__ = "restructuredtext en"

#-----------------------------------------------------------------------------
#  Copyright (C) 2008  The IPython Development Team
#
#  Distributed under the terms of the BSD License.  The full license is in
#  the file COPYING, distributed as part of this software.
#-----------------------------------------------------------------------------

#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------

# Tell nose to skip this module
__test__ = {}

import zope.interface as zi
from twisted.trial import unittest
from IPython.testing.util import DeferredTestCase

from IPython.kernel.newserialized import \
    ISerialized, \
    IUnSerialized, \
    Serialized, \
    UnSerialized, \
    SerializeIt, \
    UnSerializeIt


#-----------------------------------------------------------------------------
# Tests
#-----------------------------------------------------------------------------

class SerializedTestCase(unittest.TestCase):

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def testSerializedInterfaces(self):

        us = UnSerialized({'a':10, 'b':range(10)})
        s = ISerialized(us)
        uss = IUnSerialized(s)
        self.assert_(ISerialized.providedBy(s))
        self.assert_(IUnSerialized.providedBy(us))
        self.assert_(IUnSerialized.providedBy(uss))
        for m in list(ISerialized):
            self.assert_(hasattr(s, m))
        for m in list(IUnSerialized):
            self.assert_(hasattr(us, m))
        for m in list(IUnSerialized):
            self.assert_(hasattr(uss, m))

    def testPickleSerialized(self):
        obj = {'a':1.45345, 'b':'asdfsdf', 'c':10000L}
        original = UnSerialized(obj)
        originalSer = ISerialized(original)
        firstData = originalSer.getData()
        firstTD = originalSer.getTypeDescriptor()
        firstMD = originalSer.getMetadata()
        self.assert_(firstTD == 'pickle')
        self.assert_(firstMD == {})
        unSerialized = IUnSerialized(originalSer)
        secondObj = unSerialized.getObject()
        for k, v in secondObj.iteritems():
            self.assert_(obj[k] == v)
        secondSer = ISerialized(UnSerialized(secondObj))
        self.assert_(firstData == secondSer.getData())
        self.assert_(firstTD == secondSer.getTypeDescriptor() )
        self.assert_(firstMD == secondSer.getMetadata())

    def testNDArraySerialized(self):
        try:
            import numpy
        except ImportError:
            pass
        else:
            a = numpy.linspace(0.0, 1.0, 1000)
            unSer1 = UnSerialized(a)
            ser1 = ISerialized(unSer1)
            td = ser1.getTypeDescriptor()
            self.assert_(td == 'ndarray')
            md = ser1.getMetadata()
            self.assert_(md['shape'] == a.shape)
            self.assert_(md['dtype'] == a.dtype.str)
            buff = ser1.getData()
            self.assert_(buff == numpy.getbuffer(a))
            s = Serialized(buff, td, md)
            us = IUnSerialized(s)
            final = us.getObject()
            self.assert_(numpy.getbuffer(a) == numpy.getbuffer(final))
            self.assert_(a.dtype.str == final.dtype.str)
            self.assert_(a.shape == final.shape)