fea19ef103bfdae10a268c7c1b3f63a9e1a3701f,onmt/tests/test_preprocess.py,,,#,106

Before Change

    _add_test(p, "dataset_build")

// Test image preprocessing
for p in copy.deepcopy(test_databuild):
    p.append(("data_type", "img"))
    p.append(("src_dir", "/tmp/im2text/images"))
    p.append(("train_src", "/tmp/im2text/src-train-head.txt"))

After Change

                  [("shard_size", 50)],
                  [("tgt_vocab", "/tmp/tgt_vocab.txt")],
                  ]
test_databuild_common = [("data_type", "img"),
                         ("src_dir", "/tmp/im2text/images"),
                         ("train_src", "/tmp/im2text/src-train-head.txt"),
                         ("train_tgt", "/tmp/im2text/tgt-train-head.txt"),
                         ("valid_src", "/tmp/im2text/src-val-head.txt"),
                         ("valid_tgt", "/tmp/im2text/tgt-val-head.txt"),
                         ]
for p in test_databuild:
    _add_test(p + test_databuild_common, "dataset_build")

// Test audio preprocessing

Instances