def test_clear_series(self): values = [[0, 'yhoo', 23.0], [1, 'goog', 200]] series = Series('bar', 'symbol', 'close') series.from_values(values) self.assertEquals(len(series), 2) series.clear() self.assertEquals(len(series), 0)
def test_lol_single(self): values = [[0, 'yhoo', 23.0]] series = Series('0', '1', '2') series.from_values(values) self.assertEquals(series._0, [0]) self.assertEquals(series._1, ['yhoo']) self.assertEquals(series._2, [23.0]) self.assertEquals(len(series), 1)
def test_lol_multi(self): series = Series('0', '1', '2') values = [[0, 'yhoo', 23.0], [1, 'goog', 200]] series.from_values(values) self.assertEquals(series._0, [0, 1]) self.assertEquals(series._1, ['yhoo', 'goog']) self.assertEquals(series._2, [23.0, 200]) self.assertEquals(len(series), 2)
def test_lol_kwargs_all(self): series = Series('bar', 'symbol', 'close') values = [[0, 'yhoo', 23.0], [1, 'goog', 200]] series.from_values(values, 'bar', 'symbol') self.assertEquals(series.bar, [0, 1]) self.assertEquals(series.symbol, ['yhoo', 'goog']) self.assertEquals(series.close, [None, None]) self.assertEquals(len(series), 2)
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_lol_args_extra(self): series = Series('bar', 'symbol', 'close', 'open') values = [[0, 'yhoo', 23.0], [1, 'goog', 200]] series.from_values(values) self.assertEquals(series.bar, [0, 1]) self.assertEquals(series.symbol, ['yhoo', 'goog']) self.assertEquals(series.close, [23.0, 200]) self.assertEquals(series.open, [None, None]) self.assertEquals(len(series), 2)
def test_lod_single(self): values = [dict(bar=0, symbol='yhoo', close=23.0)] series = Series('bar', 'symbol', 'close') series.from_values(values) self.assertEquals(series.keys(), ['bar', 'symbol', 'close']) self.assertEquals(series.bar, [0]) self.assertEquals(series.symbol, ['yhoo']) self.assertEquals(series.close, [23.0]) self.assertEquals(len(series), 1)
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_sort_descending(self): series = Series('bar', 'symbol', 'close') values = [[0, 'yhoo', 23.0], [1, 'goog', 200]] series.from_values(values) series.sort('close', order='d') self.assertEquals(series.bar, [1, 0]) self.assertEquals(series.symbol, ['goog', 'yhoo']) self.assertEquals(series.close, [200, 23.0]) 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_lod_kwargs(self): values = [dict(bar=0, symbol='yhoo', close=23.0)] values.append(dict(bar=1, symbol='goog', close=200)) series = Series('bar', 'symbol', 'close') series.from_values(values, bar='bar', close='close') self.assertEquals(series.keys(), ['bar', 'symbol', 'close']) self.assertEquals(series.bar, [0, 1]) self.assertEquals(series.symbol, [None, None]) self.assertEquals(series.close, [23.0, 200]) self.assertEquals(len(series), 2)
def test_append_generator(self): values = [[0, 'yhoo', 23.0], [1, 'goog', 200]] newvalues = (x for x in values) series = Series('bar', 'symbol', 'close') series.from_values(values) self.assertEquals(series.keys(), ['bar', 'symbol', 'close']) self.assertEquals(series[0], (0, 'yhoo', 23.0)) self.assertEquals(series[1], (1, 'goog', 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_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)
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_format_int(self): values = [[0, 'yhoo', 23.0], [1, 'goog', 200]] series = Series('bar', 'symbol', 'close') series.from_values(values) series.format('close', int) self.assertEquals(series.close, [23, 200])
def test_format_str(self): values = [[0, 'yhoo', 23.0], [1, 'goog', 200]] series = Series('bar', 'symbol', 'close') series.from_values(values) series.format('close', str) self.assertEquals(series.close, ['23.0', '200'])