From 1214936f2d51ccfb12457d9a49ffbb4926216f1a Mon Sep 17 00:00:00 2001
From: Lionel GUEZ <guez@lmd.ens.fr>
Date: Thu, 17 Sep 2020 18:41:43 +0200
Subject: [PATCH] Polish
Preparing for extraction of selection task into a function.
---
Analysis/plot_snapshot.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/Analysis/plot_snapshot.py b/Analysis/plot_snapshot.py
index bdabff5f..b1c1d373 100755
--- a/Analysis/plot_snapshot.py
+++ b/Analysis/plot_snapshot.py
@@ -33,13 +33,13 @@ def snapshot(k, ax, k1, ishape_last, reader_extr, reader_outer,
dashed: boolean."""
+ assert k >= k1
+
if k == k1:
ishape_first = 0
- elif k > k1:
- ishape_first = ishape_last[k - k1 - 1] + 1
else:
- print("bad k value")
- return
+ # k > k1
+ ishape_first = ishape_last[k - k1 - 1] + 1
ax.set_title(f"k = {k}", y = 1.05)
--
GitLab