#include <migraph/program.hpp>
#include <migraph/instruction.hpp>
#include <test.hpp>
#include <basic_ops.hpp>

TEST_CASE(simple_alias)
{
    migraph::program p;
    auto l  = p.add_literal(1);
    auto p1 = p.add_instruction(pass_op{}, l);
    EXPECT(bool{migraph::instruction::get_output_alias(l) == l});
    EXPECT(bool{migraph::instruction::get_output_alias(p1) == l});
}

TEST_CASE(cascade_alias)
{
    migraph::program p;
    auto l  = p.add_literal(1);
    auto p1 = p.add_instruction(pass_op{}, l);
    auto p2 = p.add_instruction(pass_op{}, p1);
    auto p3 = p.add_instruction(pass_op{}, p2);
    EXPECT(bool{migraph::instruction::get_output_alias(l) == l});
    EXPECT(bool{migraph::instruction::get_output_alias(p1) == l});
    EXPECT(bool{migraph::instruction::get_output_alias(p2) == l});
    EXPECT(bool{migraph::instruction::get_output_alias(p3) == l});
}

TEST_CASE(no_alias)
{
    migraph::program p;
    auto x   = p.add_literal(1);
    auto y   = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, x, y);
    EXPECT(bool{migraph::instruction::get_output_alias(sum) == sum});
}

int main(int argc, const char* argv[]) { test::run(argc, argv); }