From d7ce47c13f4844c6fe5f89e25fe9a005cb525f86 Mon Sep 17 00:00:00 2001 From: ldam <ldam@passcal.nmt.edu> Date: Tue, 26 Sep 2023 16:24:42 -0600 Subject: [PATCH] flake8, type hint --- .../view/plotting/plotting_widget/plotting.py | 41 ++++++++++--------- 1 file changed, 21 insertions(+), 20 deletions(-) diff --git a/sohstationviewer/view/plotting/plotting_widget/plotting.py b/sohstationviewer/view/plotting/plotting_widget/plotting.py index 129772d5a..f0249f0f5 100644 --- a/sohstationviewer/view/plotting/plotting_widget/plotting.py +++ b/sohstationviewer/view/plotting/plotting_widget/plotting.py @@ -1,5 +1,5 @@ # class with all plotting functions -from typing import Dict, Optional +from typing import Dict import numpy as np from matplotlib.axes import Axes @@ -100,6 +100,24 @@ class Plotting: ax.chan_db_info = chan_db_info return ax + def plot_multi_color_dots_equal_on_upper_bound( + self, c_data: Dict, chan_db_info: Dict, chan_id: str) -> Axes: + """ + Use plot_multi_color_dots_base() to plot channel in which colors are + identified by plot_from_value_color_equal_on_upper_bound + """ + return self.plot_multi_color_dots_base( + c_data, chan_db_info, equal_upper=True) + + def plot_multi_color_dots_equal_on_lower_bound( + self, c_data: Dict, chan_db_info: Dict, chan_id: str) -> Axes: + """ + Use plot_multi_color_dots_base() to plot channel in which colors are + identified by plot_from_value_color_equal_on_lower_bound + """ + return self.plot_multi_color_dots_base( + c_data, chan_db_info, equal_upper=False) + def plot_tri_colors( self, c_data: Dict, chan_db_info: Dict, chan_id: str) -> Axes: """ @@ -163,25 +181,8 @@ class Plotting: ax.chan_db_info = chan_db_info return ax - def plot_multi_color_dots_equal_on_upper_bound( - self, c_data, chan_db_info, chan_id): - """ - Use plot_multi_color_dots_base() to plot channel in which colors are - identified by plot_from_value_color_equal_on_upper_bound - """ - return self.plot_multi_color_dots_base( - c_data, chan_db_info, equal_upper=True) - - def plot_multi_color_dots_equal_on_lower_bound( - self, c_data, chan_db_info, chan_id): - """ - Use plot_multi_color_dots_base() to plot channel in which colors are - identified by plot_from_value_color_equal_on_lower_bound - """ - return self.plot_multi_color_dots_base( - c_data, chan_db_info, equal_upper=False) - - def plot_up_down_dots(self, c_data, chan_db_info, chan_id): + def plot_up_down_dots( + self, c_data: Dict, chan_db_info: Dict, chan_id: str) -> Axes: """ Plot channel with 2 different values, one above, one under center line. Each value has corresponding color defined in valueColors in database. -- GitLab