From da1c70936af58fda9e0bf55d3f0353144ce9d5ec Mon Sep 17 00:00:00 2001 From: Lars Gebraad Date: Thu, 22 Mar 2018 14:59:28 +0100 Subject: [PATCH] Implented IndexErrors for all index functions. --- discretizationUtils.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/discretizationUtils.py b/discretizationUtils.py index 8f6cc0c..96a852f 100644 --- a/discretizationUtils.py +++ b/discretizationUtils.py @@ -4,7 +4,6 @@ def index2to1(y, x): - # TODO implement warnings outside of range nx = config.nx ny = config.ny @@ -15,14 +14,22 @@ def index2to1(y, x): def index1to2y(k): - # TODO implement warnings outside of range ny = config.ny + nx = config.nx + + if (k > nx * ny - 1 or k < 0): + raise (IndexError) + return int(k % ny) def index1to2x(k): - # TODO implement warnings outside of range ny = config.ny + nx = config.nx + + if (k > nx * ny - 1 or k < 0): + raise (IndexError) + return int(k / ny)