File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
tests/system/small/operations Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 12
12
# See the License for the specific language governing permissions and
13
13
# limitations under the License.
14
14
15
+ from matplotlib .collections import PathCollection
15
16
import numpy as np
16
17
import pandas as pd
17
18
import pandas ._testing as tm
@@ -258,9 +259,10 @@ def test_scatter_args_s(s):
258
259
259
260
ax = df .plot .scatter (x = "a" , y = "b" , s = "s" )
260
261
pd_ax = pd_df .plot .scatter (x = "a" , y = "b" , s = "s" )
261
- # TODO(b/340891723): fix type error
262
+
263
+ assert isinstance (pd_ax .collections [0 ], PathCollection )
262
264
tm .assert_numpy_array_equal (
263
- ax .collections [0 ].get_sizes (), pd_ax .collections [0 ].get_sizes () # type: ignore
265
+ ax .collections [0 ].get_sizes (), pd_ax .collections [0 ].get_sizes ()
264
266
)
265
267
266
268
You can’t perform that action at this time.
0 commit comments