From b7dc344c9e60766d0cc7fe21d726f731fb9d1475 Mon Sep 17 00:00:00 2001
From: Rafal Skolasinski <r.j.skolasinski@gmail.com>
Date: Tue, 27 Jun 2017 16:21:20 +0200
Subject: [PATCH] make use of "as_ordered_terms" in "_extract_hoppings"
---
kwant/continuum/discretizer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/kwant/continuum/discretizer.py b/kwant/continuum/discretizer.py
index 44788ecd..5f8705b6 100644
--- a/kwant/continuum/discretizer.py
+++ b/kwant/continuum/discretizer.py
@@ -405,7 +405,7 @@ def _discretize_expression(expression, coords):
def _extract_hoppings(expr):
"""Read hoppings and perform shortening operation."""
expr = sympy.expand(expr)
- summands = expr.args if expr.func == sympy.Add else [expr]
+ summands = expr.as_ordered_terms()
offset = [_read_offset(s.args[-1]) for s in summands]
coeffs = [sympy.Mul(*s.args[:-1]) for s in summands]
--
GitLab