From d61e6d4f17afe7b80153e71f4f4d734a83d9b5cf Mon Sep 17 00:00:00 2001 From: Lorenz Kies Date: Sun, 2 Mar 2025 22:49:33 +0100 Subject: [PATCH 1/2] fix color cycling not working in singular_values_plot this seems to have ben caused by a name collision between the function scope color variable and the loop-"local" color variable. the first time _get_color is called it will replace the function level color so in the next iteration a color is explicitly passed to _get_color so it will no longer automatically cycle through colors. --- control/freqplot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/control/freqplot.py b/control/freqplot.py index fe258d636..048979960 100644 --- a/control/freqplot.py +++ b/control/freqplot.py @@ -2556,12 +2556,12 @@ def singular_values_plot( nyq_freq = None # Determine the color to use for this response - color = _get_color( + current_color = _get_color( color, fmt=fmt, offset=color_offset + idx_sys, color_cycle=color_cycle) # To avoid conflict with *fmt, only pass color kw if non-None - color_arg = {} if color is None else {'color': color} + color_arg = {} if current_color is None else {'color': current_color} # Decide on the system name sysname = response.sysname if response.sysname is not None \ From 049a71657fb348214dd4972d9b53183389a2240a Mon Sep 17 00:00:00 2001 From: Lorenz Kies Date: Sun, 2 Mar 2025 23:21:06 +0100 Subject: [PATCH 2/2] test to verify that color cycling works for singular_values_plot --- control/tests/freqplot_test.py | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) diff --git a/control/tests/freqplot_test.py b/control/tests/freqplot_test.py index c0dfa4030..b3770486c 100644 --- a/control/tests/freqplot_test.py +++ b/control/tests/freqplot_test.py @@ -680,6 +680,39 @@ def test_display_margins(nsys, display_margins, gridkw, match): assert cplt.axes[0, 0].get_title() == '' +def test_singular_values_plot_colors(): + # Define some systems for testing + sys1 = ct.rss(4, 2, 2, strictly_proper=True) + sys2 = ct.rss(4, 2, 2, strictly_proper=True) + + # Get the default color cycle + color_cycle = plt.rcParams['axes.prop_cycle'].by_key()['color'] + + # Plot the systems individually and make sure line colors are OK + cplt = ct.singular_values_plot(sys1) + assert cplt.lines.size == 1 + assert len(cplt.lines[0]) == 2 + assert cplt.lines[0][0].get_color() == color_cycle[0] + assert cplt.lines[0][1].get_color() == color_cycle[0] + + cplt = ct.singular_values_plot(sys2) + assert cplt.lines.size == 1 + assert len(cplt.lines[0]) == 2 + assert cplt.lines[0][0].get_color() == color_cycle[1] + assert cplt.lines[0][1].get_color() == color_cycle[1] + plt.close('all') + + # Plot the systems as a list and make sure colors are OK + cplt = ct.singular_values_plot([sys1, sys2]) + assert cplt.lines.size == 2 + assert len(cplt.lines[0]) == 2 + assert len(cplt.lines[1]) == 2 + assert cplt.lines[0][0].get_color() == color_cycle[0] + assert cplt.lines[0][1].get_color() == color_cycle[0] + assert cplt.lines[1][0].get_color() == color_cycle[1] + assert cplt.lines[1][1].get_color() == color_cycle[1] + + if __name__ == "__main__": # # Interactive mode: generate plots for manual viewing pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.

Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy