Skip to content
Snippets Groups Projects
Commit 941be0b8 authored by Lionel GUEZ's avatar Lionel GUEZ
Browse files

Check argument label in function `is_node_type`

parent e35226da
No related branches found
No related tags found
No related merge requests found
......@@ -7,6 +7,7 @@ import numpy as np
from matplotlib import patches
import cartopy.crs as ccrs
import report_graph
import sys
color_iter = itertools.cycle(('#1f77b4', '#aec7e8', '#ff7f0e',
'#ffbb78', '#2ca02c', '#98df8a',
......@@ -40,6 +41,8 @@ def is_node_type(G, n, label):
only_date = True) == label[1]
elif label[0] == "node":
return n in label[1]
else:
sys.exit(f"is_node_type: bad label: {label}")
def plot_nbunch(G, nbunch, color = '#1f78b4', label = None, ax = None):
if ax is None: ax = plt.gca()
......@@ -96,7 +99,6 @@ def animate_nbunch(G, nbunch):
if __name__ == "__main__":
import argparse
import sys
from os import path
sys.path.append(path.join(sys.path[0], "../Common"))
import util_eddies
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment