1# Copyright 2020 Huawei Technologies Co., Ltd 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14# ============================================================================== 15""" 16Testing Duplicate op in DE 17""" 18import numpy as np 19 20import mindspore.dataset as ds 21import mindspore.dataset.transforms.c_transforms as ops 22 23 24def compare(array): 25 data = ds.NumpySlicesDataset([array], column_names="x") 26 array = np.array(array) 27 data = data.map(operations=ops.Duplicate(), input_columns=["x"], output_columns=["x", "y"], 28 column_order=["x", "y"]) 29 for d in data.create_dict_iterator(num_epochs=1, output_numpy=True): 30 np.testing.assert_array_equal(array, d["x"]) 31 np.testing.assert_array_equal(array, d["y"]) 32 33 34def test_duplicate_basics(): 35 compare([1, 2, 3]) 36 compare([b"1", b"2", b"3"]) 37 38 39if __name__ == "__main__": 40 test_duplicate_basics() 41