Added missing unittest for WafermapDataset