def METHOD_NAME(self):
    fig = go.Figure(data=go.Waterfall(y=[2, 1, 3]))
    self.assertEqual(fig.data, (go.Waterfall(y=[2, 1, 3]),))
    fig = go.Figure(data=dict(type="waterfall", y=[2, 1, 3]))
    self.assertEqual(fig.data, (go.Waterfall(y=[2, 1, 3]),))