def testDataProviderWithOverwrittern(self):
dataSource = DataProvider()
dataSource.set('aapl', 'close', 2.0)
with self.assertRaises(ValueError):
dataSource.set('aapl', 'close', 3.0)
def testDataProviderWithOverwrittern(self):
dataSource = DataProvider()
dataSource.set('aapl', 'close', 2.0)
with self.assertRaises(ValueError):
dataSource.set('aapl', 'close', 3.0)
def testDataProvider(self):
dataSource = DataProvider()
dataSource.set('aapl', 'close', 2.0)
dataSource.set('ibm', 'open', 3.0)
dataSource.set('goog', 'pe', 4.0)
dataSource.set('aapl', 'rsi', 5.0)
calculated = dataSource.data
expected = {
'aapl': {
'close': 2.0,
'rsi': 5.0
},
'ibm': {
'open': 3.0
},
'goog': {
'pe': 4.0
}
}
for name in expected:
for field in expected[name]:
eres = expected[name][field]
cres = calculated[name][field]
self.assertEqual(
eres, cres, "for {0}'s field {1}\n"
"expected: {2}\n"
"calculated: {3}".format(name, field, eres, cres))
dataSource.clear()
calculated = dataSource.data
expected = {}
self.assertEqual(
calculated, expected, "cleared data provide\n"
"expected: {0}\n"
"calculated: {1}".format(expected, calculated))
def testDataProvider(self):
dataSource = DataProvider()
dataSource.set('aapl', 'close', 2.0)
dataSource.set('ibm', 'open', 3.0)
dataSource.set('goog', 'pe', 4.0)
dataSource.set('aapl', 'rsi', 5.0)
calculated = dataSource.data
expected = {
'aapl': {'close': 2.0, 'rsi': 5.0},
'ibm': {'open': 3.0},
'goog': {'pe': 4.0}
}
for name in expected:
for field in expected[name]:
eres = expected[name][field]
cres = calculated[name][field]
self.assertEqual(eres, cres, "for {0}'s field {1}\n"
"expected: {2}\n"
"calculated: {3}".format(name, field, eres, cres))
dataSource.clear()
calculated = dataSource.data
expected = {}
self.assertEqual(calculated, expected, "cleared data provide\n"
"expected: {0}\n"
"calculated: {1}".format(expected, calculated))
