def METHOD_NAME(self):
    ds = DatasetFromList([1, 2, 3])
    ds = MapDataset(ds, TestMapDataset.map_func)
    self.assertEqual(ds[0], 2)
    self.assertEqual(ds[2], 6)
    self.assertIn(ds[1], [2, 6])