def test_fake_tensor(): ''' Blends have a fake tensor that does iteritems really fast. ''' # labels will be abc, 12 t3 = ez_matrix('c', '1', [1]) b = Blend([t1, t3], weights=[.75, .25]) fake_tensor = b.fake_tensor() items = set(fake_tensor.iteritems()) expected_items = set( (((0, 0), .75), ((0, 1), .75), ((1, 0), .75), ((2, 0), .25))) assertSetEquals(items, expected_items)
def test_fake_tensor(): ''' Blends have a fake tensor that does iteritems really fast. ''' # labels will be abc, 12 t3 = ez_matrix('c', '1', [1]) b = Blend([t1, t3], weights=[.75,.25]) fake_tensor = b.fake_tensor() items = set(fake_tensor.iteritems()) expected_items = set(( ((0,0), .75), ((0,1), .75), ((1,0), .75), ((2,0), .25))) assertSetEquals(items, expected_items)