Some_unittest.cc

#include <gtest/gtest.h>
#include <shogun/base/SGObject.h>
#include <shogun/base/some.h>
#include <shogun/classifier/AveragedPerceptron.h>
#include <shogun/kernel/GaussianKernel.h>

#ifdef HAVE_CXX11
using namespace shogun;

TEST(Some,basic)
{
    // raw pointer to the kernel
    CKernel* raw = NULL;
    // local scope to create kernel
    {
        auto kernel = some<CGaussianKernel>();
        EXPECT_EQ(1, kernel->ref_count());
        EXPECT_EQ(1, kernel->ref_count());
        raw = kernel;
        SG_REF(raw);
        EXPECT_TRUE(kernel->equals(raw));

        // reference is held
        EXPECT_EQ(2, kernel->ref_count());
    }

	EXPECT_TRUE(raw);
	// last references now
	EXPECT_EQ(1, raw->ref_count());
	SG_UNREF(raw);
}

TEST(Some,reassignment)
{
    auto kernel = some<CGaussianKernel>();
    CGaussianKernel* raw = new CGaussianKernel();
    EXPECT_EQ(1, kernel->ref_count());
    EXPECT_EQ(0, raw->ref_count());
	kernel.reset(raw);
	EXPECT_TRUE(kernel->equals(raw));
	EXPECT_EQ(0, kernel->ref_count());
}

TEST(Some,self_assignment)
{
    auto kernel = some<CGaussianKernel>();
    kernel = kernel;
    EXPECT_EQ(1, kernel->ref_count());
}

TEST(Some,get)
{
    auto kernel = some<CGaussianKernel>();
    CGaussianKernel* raw = kernel;
    SG_REF(raw);
    EXPECT_TRUE(kernel->equals(raw));
    EXPECT_EQ(2, raw->ref_count());
    SG_UNREF(raw);
    EXPECT_EQ(1, raw->ref_count());
}

TEST(Some, get_method)
{
	auto raw = new CGaussianKernel();
	SG_REF(raw);
	auto kernel = wrap(raw);
	EXPECT_TRUE(raw == kernel.get());
	EXPECT_EQ(2, raw->ref_count());
	SG_UNREF(raw);
	EXPECT_EQ(1, raw->ref_count());
}

TEST(Some, constructor_new_type)
{
	auto kernel = some<CGaussianKernel>();
	EXPECT_EQ(1, kernel->ref_count());
	auto object = Some<CSGObject>(kernel);
	EXPECT_EQ(2, kernel->ref_count());
}

TEST(Some, constructor_new_type_wrong_casting)
{
	auto kernel = some<CGaussianKernel>();
	auto object = Some<CAveragedPerceptron>(kernel);
	EXPECT_EQ(1, kernel->ref_count());
	EXPECT_TRUE(object.get() == nullptr);
	EXPECT_TRUE(kernel.get() != nullptr);
}

#endif  // HAVE_CXX11