literal_test.cpp

#include <migraph/literal.hpp>
#include <sstream>
#include <string>
#include "test.hpp"

TEST_CASE(literal_test)
{
    EXPECT(migraph::literal{1} == migraph::literal{1});
    EXPECT(migraph::literal{1} != migraph::literal{2});
    EXPECT(migraph::literal{} == migraph::literal{});
    EXPECT(migraph::literal{} != migraph::literal{2});

    migraph::literal l1{1};
    migraph::literal l2 = l1; // NOLINT
    EXPECT(l1 == l2);
    EXPECT(l1.at<int>(0) == 1);
    EXPECT(!l1.empty());
    EXPECT(!l2.empty());

    migraph::literal l3{};
    migraph::literal l4{};
    EXPECT(l3 == l4);
    EXPECT(l3.empty());
    EXPECT(l4.empty());
}

TEST_CASE(literal_os1)
{
    migraph::literal l{1};
    std::stringstream ss;
    ss << l;
    EXPECT(ss.str() == "1");
}

TEST_CASE(literal_os2)
{
    migraph::literal l{};
    std::stringstream ss;
    ss << l;
    EXPECT(ss.str().empty());
}

TEST_CASE(literal_os3)
{
    migraph::shape s{migraph::shape::int64_type, {3}};
    migraph::literal l{s, {1, 2, 3}};
    std::stringstream ss;
    ss << l;
    EXPECT(ss.str() == "1, 2, 3");
}

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