60 self.assertEqual(values[0].rtype, 'total_salary') |
60 self.assertEqual(values[0].rtype, 'total_salary') |
61 |
61 |
62 def test_computed_attribute_by_relation(self): |
62 def test_computed_attribute_by_relation(self): |
63 comp_by_rdef = self.dependencies.computed_attribute_by_relation |
63 comp_by_rdef = self.dependencies.computed_attribute_by_relation |
64 self.assertEqual(len(comp_by_rdef), 1) |
64 self.assertEqual(len(comp_by_rdef), 1) |
65 key, values = iter(comp_by_rdef.iteritems()).next() |
65 key, values = iter(comp_by_rdef.items()).next() |
66 self.assertEqual(key.rtype, 'works_for') |
66 self.assertEqual(key.rtype, 'works_for') |
67 self.assertEqual(len(values), 1) |
67 self.assertEqual(len(values), 1) |
68 self.assertEqual(values[0].rtype, 'total_salary') |
68 self.assertEqual(values[0].rtype, 'total_salary') |
69 |
69 |
70 def test_computed_attribute_by_etype_attrs(self): |
70 def test_computed_attribute_by_etype_attrs(self): |
71 comp_by_attr = self.dependencies.computed_attribute_by_etype_attrs |
71 comp_by_attr = self.dependencies.computed_attribute_by_etype_attrs |
72 self.assertEqual(len(comp_by_attr), 1) |
72 self.assertEqual(len(comp_by_attr), 1) |
73 values = comp_by_attr['Person'] |
73 values = comp_by_attr['Person'] |
74 self.assertEqual(len(values), 2) |
74 self.assertEqual(len(values), 2) |
75 values = set((rdef.formula, tuple(v)) |
75 values = set((rdef.formula, tuple(v)) |
76 for rdef, v in values.iteritems()) |
76 for rdef, v in values.items()) |
77 self.assertEquals(values, |
77 self.assertEquals(values, |
78 set((('Any 2014 - D WHERE X birth_year D', tuple(('birth_year',))), |
78 set((('Any 2014 - D WHERE X birth_year D', tuple(('birth_year',))), |
79 ('Any SUM(SA) GROUPBY X WHERE P works_for X, P salary SA', tuple(('salary',))))) |
79 ('Any SUM(SA) GROUPBY X WHERE P works_for X, P salary SA', tuple(('salary',))))) |
80 ) |
80 ) |
81 |
81 |