mirror of
https://github.com/wassname/catalyst.git
synced 2026-06-28 14:47:08 +08:00
fawce
63 lines
1.3 KiB
Python
63 lines
1.3 KiB
Python
from datetime import datetime
|
|
import pytz
|
|
|
|
from zipline.utils.protocol_utils import ndict
|
|
|
|
def test_ndict():
|
|
nd = ndict({})
|
|
|
|
# Properties
|
|
assert len(nd) == 0
|
|
assert nd.keys() == []
|
|
assert nd.values() == []
|
|
assert list(nd.iteritems()) == []
|
|
|
|
# Accessors
|
|
nd['x'] = 1
|
|
assert nd.x == 1
|
|
assert nd['x'] == 1
|
|
assert nd.get('y') == None
|
|
assert nd.get('y', 'fizzpop') == 'fizzpop'
|
|
assert nd.has_key('x') == True
|
|
assert nd.has_key('y') == False
|
|
|
|
assert 'x' in nd
|
|
assert 'y' not in nd
|
|
|
|
# Mutability
|
|
nd2 = ndict({'x': 1})
|
|
assert nd2.x == 1
|
|
nd2.x = 2
|
|
assert nd2.x == 2
|
|
|
|
# Class isolation
|
|
assert '__init__' not in nd
|
|
assert '__iter__' not in nd
|
|
assert not nd.__dict__.has_key('x')
|
|
assert nd.get('__init__') is None
|
|
assert 'x' not in set(dir(nd))
|
|
|
|
# Comparison
|
|
nd2 = nd.copy()
|
|
assert id(nd2) != id(nd)
|
|
assert nd2 == nd
|
|
nd2['z'] = 3
|
|
assert nd2 != nd
|
|
|
|
class ndictlike(object):
|
|
x = 1
|
|
|
|
assert { 'x': 1 } == nd
|
|
assert ndictlike() != nd
|
|
|
|
# Deletion
|
|
del nd['x']
|
|
assert not nd.has_key('x')
|
|
assert nd.get('x') is None
|
|
|
|
|
|
for n in xrange(1000):
|
|
dt = datetime.utcnow().replace(tzinfo=pytz.utc)
|
|
nd2 = ndict({"dt":dt, "otherdata":"ishere"*1000, "maybeanint":3})
|
|
|
|
nd2.dt2 = dt |