Skip to content

Commit d5ae680

Browse files
authored
chore: cleanup type errors in tests/system/small/operations/test_plotting.py (#785)
1 parent e14c7a9 commit d5ae680

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/system/small/operations/test_plotting.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
from matplotlib.collections import PathCollection
1516
import numpy as np
1617
import pandas as pd
1718
import pandas._testing as tm
@@ -258,9 +259,10 @@ def test_scatter_args_s(s):
258259

259260
ax = df.plot.scatter(x="a", y="b", s="s")
260261
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)
262264
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()
264266
)
265267

266268

0 commit comments

Comments
 (0)