TST: add test for plot with nans
BUG: make plot work when data contains nans
add comment about nans