eval/src/apps/tensor_conformance/generate.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#pragma once

#include <vespa/eval/eval/tensor_spec.h>
#include <map>

struct TestBuilder {
    using TensorSpec = vespalib::eval::TensorSpec;
    // add test with pre-defined expected result
    virtual void add(const vespalib::string &expression,
                     const std::map<vespalib::string,TensorSpec> &inputs,
                     const TensorSpec &expect) = 0;
    // add test with undefined expected result
    virtual void add(const vespalib::string &expression,
                     const std::map<vespalib::string,TensorSpec> &inputs) = 0;
    virtual ~TestBuilder() {}
};

struct Generator {
    static void generate(TestBuilder &out);
};