From 89e1cc33f2a8a5c1d940c19ba7f6c71652948374 Mon Sep 17 00:00:00 2001 From: Anton Khirnov Date: Wed, 22 Oct 2014 14:57:55 +0200 Subject: series_expansion: allow passing scalar diff_order to 1-dim expansions --- series_expansion.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/series_expansion.py b/series_expansion.py index edee84e..76adc30 100644 --- a/series_expansion.py +++ b/series_expansion.py @@ -44,6 +44,12 @@ class SeriesExpansion(object): except (TypeError, IndexError): coords = [coords] + if diff_order is not None: + try: + diff_order[0] + except (TypeError, IndexError): + diff_order = [diff_order] + if diff_order is None: diff_order = [0] * len(coords) -- cgit v1.2.3