mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-20 02:55:47 +00:00
91 lines
3.2 KiB
C++
91 lines
3.2 KiB
C++
//============================================================================
|
|
// Copyright (c) Kitware, Inc.
|
|
// All rights reserved.
|
|
// See LICENSE.txt for details.
|
|
//
|
|
// This software is distributed WITHOUT ANY WARRANTY; without even
|
|
// the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
// PURPOSE. See the above copyright notice for more information.
|
|
//============================================================================
|
|
#include <vtkm/cont/ArrayHandleCounting.h>
|
|
|
|
#include <vtkm/cont/testing/Testing.h>
|
|
|
|
#include <string>
|
|
|
|
namespace UnitTestArrayHandleCountingNamespace
|
|
{
|
|
|
|
const vtkm::Id ARRAY_SIZE = 10;
|
|
|
|
} // namespace UnitTestArrayHandleCountingNamespace
|
|
|
|
namespace UnitTestArrayHandleCountingNamespace
|
|
{
|
|
|
|
template <typename ValueType>
|
|
struct TemplatedTests
|
|
{
|
|
using ArrayHandleType = vtkm::cont::ArrayHandleCounting<ValueType>;
|
|
|
|
using ArrayHandleType2 = vtkm::cont::ArrayHandle<ValueType, vtkm::cont::StorageTagCounting>;
|
|
|
|
using PortalType =
|
|
typename vtkm::cont::internal::Storage<ValueType,
|
|
typename ArrayHandleType::StorageTag>::ReadPortalType;
|
|
|
|
void operator()(const ValueType& startingValue, const ValueType& step)
|
|
{
|
|
ArrayHandleType arrayConst(startingValue, step, ARRAY_SIZE);
|
|
|
|
ArrayHandleType arrayMake =
|
|
vtkm::cont::make_ArrayHandleCounting(startingValue, step, ARRAY_SIZE);
|
|
|
|
ArrayHandleType2 arrayHandle = ArrayHandleType(startingValue, step, ARRAY_SIZE);
|
|
|
|
VTKM_TEST_ASSERT(arrayConst.GetNumberOfValues() == ARRAY_SIZE,
|
|
"Counting array using constructor has wrong size.");
|
|
|
|
VTKM_TEST_ASSERT(arrayMake.GetNumberOfValues() == ARRAY_SIZE,
|
|
"Counting array using make has wrong size.");
|
|
|
|
VTKM_TEST_ASSERT(arrayHandle.GetNumberOfValues() == ARRAY_SIZE,
|
|
"Counting array using raw array handle + tag has wrong size.");
|
|
|
|
ValueType properValue = startingValue;
|
|
auto arrayConstPortal = arrayConst.ReadPortal();
|
|
auto arrayMakePortal = arrayMake.ReadPortal();
|
|
auto arrayHandlePortal = arrayHandle.ReadPortal();
|
|
for (vtkm::Id index = 0; index < ARRAY_SIZE; index++)
|
|
{
|
|
VTKM_TEST_ASSERT(arrayConstPortal.Get(index) == properValue,
|
|
"Counting array using constructor has unexpected value.");
|
|
VTKM_TEST_ASSERT(arrayMakePortal.Get(index) == properValue,
|
|
"Counting array using make has unexpected value.");
|
|
|
|
VTKM_TEST_ASSERT(arrayHandlePortal.Get(index) == properValue,
|
|
"Counting array using raw array handle + tag has unexpected value.");
|
|
properValue = properValue + step;
|
|
}
|
|
}
|
|
};
|
|
|
|
void TestArrayHandleCounting()
|
|
{
|
|
TemplatedTests<vtkm::Id>()(0, 1);
|
|
TemplatedTests<vtkm::Id>()(8, 2);
|
|
TemplatedTests<vtkm::Float32>()(0.0f, 1.0f);
|
|
TemplatedTests<vtkm::Float32>()(3.0f, -0.5f);
|
|
TemplatedTests<vtkm::Float64>()(0.0, 1.0);
|
|
TemplatedTests<vtkm::Float64>()(-3.0, 2.0);
|
|
}
|
|
|
|
|
|
} // namespace UnitTestArrayHandleCountingNamespace
|
|
|
|
int UnitTestArrayHandleCounting(int argc, char* argv[])
|
|
{
|
|
using namespace UnitTestArrayHandleCountingNamespace;
|
|
return vtkm::cont::testing::Testing::Run(TestArrayHandleCounting, argc, argv);
|
|
}
|