Fixed the printing of rank-0 tensors

This commit is contained in:
Benoit Steiner
2016-06-20 10:46:45 -07:00
parent b055590e91
commit de32f8d656
2 changed files with 62 additions and 22 deletions

View File

@@ -13,6 +13,20 @@
#include <Eigen/CXX11/Tensor>
template<int DataLayout>
static void test_output_0d()
{
Tensor<int, 0, DataLayout> tensor;
tensor() = 123;
std::stringstream os;
os << tensor;
std::string expected("123");
VERIFY_IS_EQUAL(std::string(os.str()), expected);
}
template<int DataLayout>
static void test_output_1d()
{
@@ -101,6 +115,8 @@ static void test_output_const()
void test_cxx11_tensor_io()
{
CALL_SUBTEST(test_output_0d<ColMajor>());
CALL_SUBTEST(test_output_0d<RowMajor>());
CALL_SUBTEST(test_output_1d<ColMajor>());
CALL_SUBTEST(test_output_1d<RowMajor>());
CALL_SUBTEST(test_output_2d<ColMajor>());