transforms/lower_units: support empty list

This commit is contained in:
Sebastien Bourdeauducq 2014-12-19 14:40:20 +08:00
parent 5522378c1c
commit a9b28dff36
1 changed files with 5 additions and 4 deletions

View File

@ -89,6 +89,7 @@ class _UnitsLowerer(ast.NodeTransformer):
def visit_List(self, node): def visit_List(self, node):
self.generic_visit(node) self.generic_visit(node)
if node.elts:
us = [getattr(elt, "unit", None) for elt in node.elts] us = [getattr(elt, "unit", None) for elt in node.elts]
if not all(u == us[0] for u in us[1:]): if not all(u == us[0] for u in us[1:]):
raise units.DimensionError raise units.DimensionError