UpsilonNumworks/Upsilon
1
0
Fork 0
mirror of https://github.com/UpsilonNumworks/Upsilon.git synced 2026-01-19 08:47:28 +01:00
Code Issues Packages Projects Releases Wiki Activity

[python/matplotlib] Validate the plot parameters

Browse Source
This commit is contained in:
Romain Goyet
2020-03-11 10:36:38 -04:00
committed by Émilie Feral
parent 6a796f5f7c
commit 6f3d3f62de
3 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
python/port/mod/matplotlib/modpyplot.cpp
View File

@@ -35,7 +35,8 @@ void modpyplot_gc_collect() {
* Returns : xmin, xmax, ymin, ymax : float */
mp_obj_t modpyplot_axis(mp_obj_t arg) {
mp_obj_is_type(arg, &mp_type_enumerate);
assert(sPlotStore != nullptr);
#warning Use mp_obj_is_bool when upgrading uPy
if (mp_obj_is_type(arg, &mp_type_bool)) {
sPlotStore->setGrid(mp_obj_is_true(arg));
@@ -59,9 +60,18 @@ mp_obj_t modpyplot_axis(mp_obj_t arg) {
mp_obj_t modpyplot_plot(mp_obj_t x, mp_obj_t y) {
assert(sPlotStore != nullptr);
// Input parameter validation
size_t xLength, yLength;
mp_obj_t * xItems, yItems;
mp_obj_get_array(x, &xLength, &xItems);
mp_obj_get_array(y, &yLength, &yItems);
if (xLength != yLength) {
mp_raise_msg_varg(&mp_type_ValueError, "x and y must have same dimension");
}
sPlotStore->addDots(x, y);
// Ensure x and y are arrays
// "Push" x and y on bigger arrays
return mp_const_none;
}