class TestColorData(unittest.TestCase):
def setUp(self):
builder = MockBuilder()
blockchain = MockBlockchain()
store = MockStore()
colormap = MockColorMap()
self.thick = ThickColorData(builder, blockchain, store, colormap)
self.thin = ThinColorData(builder, blockchain, store, colormap)
def test_thick(self):
self.assertRaises(UnfoundTransactionError, self.thick.get_colorvalues,
set([1,2]), '', 0)
cvs = self.thick.get_colorvalues(set([1,2]), '1', 0)
self.assertEquals(cvs[0].get_value(), 5)
self.assertEquals(cvs[1].get_value(), 6)
cvs = self.thick.get_colorvalues(set([1,2]), '2', 0)
self.assertEquals(cvs[0].get_value(), 5)
self.assertEquals(cvs[1].get_value(), 6)
self.assertRaises(UnfoundTransactionError, self.thick.get_colorvalues,
set([1,2]), '9', 0)
def test_thin(self):
self.assertRaises(UnfoundTransactionError, self.thin.get_colorvalues,
set([1,2]), 'nope', 0)
cvs = self.thin.get_colorvalues(set([1,2]), '1', 0)
self.assertEquals(cvs[0].get_value(), 5)
self.assertEquals(cvs[1].get_value(), 6)
cvs = self.thin.get_colorvalues(set([1,2]), '2', 0)
self.assertEquals(cvs[0].get_value(), 5)
self.assertEquals(cvs[1].get_value(), 6)
