def test_project_3_dom_to_2(self):
dom = {0:2, 1:1, 2:3}
dim = {0:{0:1, 1:1}, 1:{2:3, 3:2.0}, 2:{4:1}}
w = Weights(dom, dim)
dom_res = {0:2, 2:3}
dim_res = {0:{0:1, 1:1}, 2:{4:1}}
w_res = Weights(dom_res, dim_res)
self.assertEqual(w.project_onto({0:[0,1], 2:[4]}), w_res)
def test_project_2_dom_to_1(self):
dom = {0:2, 1:1}
dim = {0:{0:1, 1:1}, 1:{2:3, 3:2.0}}
w = Weights(dom, dim)
dom_res = {0:1}
dim_res = {0:{0:1, 1:1}}
w_res = Weights(dom_res, dim_res)
self.assertEqual(w.project_onto({0:[0,1]}), w_res)
