def test_appendcol_valuesbad(self):
values = [dict(bar=0, symbol='yhoo', close=23.0)]
series = Series('bar', 'symbol', 'close')
series.from_values(values)
values = dict(bar=1, symbol='goog', close=200)
series.append(values)
self.assertRaises(ValueError, series.appendcol, 'open', [22.0])
def test_index_access(self):
values = [dict(bar=0, symbol='yhoo', close=23.0)]
series = Series('bar', 'symbol', 'close')
series.from_values(values)
values = dict(bar=1, symbol='goog', close=200)
series.append(values)
self.assertEquals(series[0], (0, 'yhoo', 23.0))
self.assertEquals(series[1], (1, 'goog', 200))
self.assertEquals(len(series), 2)
def test_append_dict_args_none(self):
values = [dict(bar=0, symbol='yhoo', close=23.0)]
series = Series('bar', 'symbol', 'close')
series.from_values(values)
series.append(dict(bar=1, symbol='goog', close=200))
self.assertEquals(series.bar, [0, 1])
self.assertEquals(series.symbol, ['yhoo', 'goog'])
self.assertEquals(series.close, [23.0, 200])
self.assertEquals(len(series), 2)
def test_append_list_args_none(self):
values = [[0, 'yhoo', 23.0]]
series = Series('bar', 'symbol', 'close')
series.from_values(values)
series.append([1, 'goog', 200])
self.assertEquals(series.bar, [0, 1])
self.assertEquals(series.symbol, ['yhoo', 'goog'])
self.assertEquals(series.close, [23.0, 200])
self.assertEquals(len(series), 2)
def test_iter(self):
values = [dict(bar=0, symbol='yhoo', close=23.0)]
series = Series('bar', 'symbol', 'close')
series.from_values(values)
values = dict(bar=1, symbol='goog', close=200)
series.append(values)
results = []
for bar in series:
results.append(bar)
self.assertEquals(results, [0, 1])
def test_appendcol_values(self):
values = [dict(bar=0, symbol='yhoo', close=23.0)]
series = Series('bar', 'symbol', 'close')
series.from_values(values)
values = dict(bar=1, symbol='goog', close=200)
series.append(values)
series.appendcol('open', [22.0, 21.0])
self.assertEquals(series.keys(), ['bar', 'symbol', 'close', 'open'])
self.assertEquals(series[0], (0, 'yhoo', 23.0, 22.0))
self.assertEquals(series[1], (1, 'goog', 200, 21.0))
self.assertEquals(len(series), 2)
def test_initcol_existing_0(self):
values = [dict(bar=0, symbol='yhoo', close=23.0)]
series = Series('bar', 'symbol', 'close')
series.from_values(values)
values = dict(bar=1, symbol='goog', close=200)
series.append(values)
series.initcol('close', 0.0)
self.assertEquals(series.keys(), ['bar', 'symbol', 'close'])
self.assertEquals(series[0], (0, 'yhoo', 0.0))
self.assertEquals(series[1], (1, 'goog', 0.0))
self.assertEquals(len(series), 2)
