def testDataProvider(self):
result = api.evaluate('2*GOOG, GOOG')
self.assertEqual(len(result.data), 1)
self.assertEqual(result.expression, api.parse('2*GOOG, GOOG'))
data = result.ts()
self.assertTrue(api.is_timeseries(data))
self.assertEqual(data.count(), 2)
ts1 = data.serie(0)
ts2 = data.serie(1)
for v1, v2 in zip(ts1, ts2):
self.assertAlmostEqual(v1, 2. * v2)
def testDataProvider(self):
result = api.evaluate('2*GOOG, GOOG')
self.assertEqual(len(result.data), 1)
self.assertEqual(result.expression, api.parse('2*GOOG, GOOG'))
data = result.ts()
self.assertTrue(api.is_timeseries(data))
self.assertEqual(data.count(), 2)
ts1 = data.serie(0)
ts2 = data.serie(1)
for v1, v2 in zip(ts1, ts2):
self.assertAlmostEqual(v1, 2.*v2)
def testBinOpSerieFunction(self):
'''Get a timeseries and a function and check for consistency'''
res = api.parse('goog:yahoo-ma(goog:yahoo, window=30)')
self.assertEqual(res.symbols(), ['GOOG:YAHOO'])
self.assertEqual(len(res), 2)
self.assertEqual(str(res[0]), 'GOOG:YAHOO')
self.assertEqual(str(res[1]), 'ma(GOOG:YAHOO, window=30)')
result = api.evaluate(res)
self.assertEqual(len(result.data), 1)
data = result.ts()
self.assertTrue(api.is_timeseries(data))
self.assertEqual(data.count(), 1)
def testBinOpSerieFunction(self):
'''Get a timeseries and a function and check for consistency'''
res = api.parse('goog:yahoo-ma(goog:yahoo, window=30)')
self.assertEqual(res.symbols(), ['GOOG:YAHOO'])
self.assertEqual(len(res), 2)
self.assertEqual(str(res[0]), 'GOOG:YAHOO')
self.assertEqual(str(res[1]), 'ma(GOOG:YAHOO, window=30)')
result = api.evaluate(res)
self.assertEqual(len(result.data), 1)
data = result.ts()
self.assertTrue(api.is_timeseries(data))
self.assertEqual(data.count(), 1)
def testTSName(self):
'''
The dslresult should include an attribute 'name'
which is the equivalent to the expression passed.
In situations where multiple timeseries are returned
the name should be the concatenation of all the names
joined by "__".
'''
expressions = ['GOOG+YHOO',
'2*GOOG',
'GOOG, YHOO',
]
for expr in expressions:
result = api.evaluate(expr)
ts = result.ts()
name = ts.name
expected_name = '__'.join(expr.split(','))
self.assertEqual(name, expected_name)
def testTSName(self):
'''
The dslresult should include an attribute 'name'
which is the equivalent to the expression passed.
In situations where multiple timeseries are returned
the name should be the concatenation of all the names
joined by "__".
'''
expressions = [
'GOOG+YHOO',
'2*GOOG',
'GOOG, YHOO',
]
for expr in expressions:
result = api.evaluate(expr)
ts = result.ts()
name = ts.name
expected_name = '__'.join(expr.split(','))
self.assertEqual(name, expected_name)
