Refs #11690 Add unit test

mantidproject · May 6, 2015 · 0b0df2c · 0b0df2c
1 parent 1f72ffa
commit 0b0df2c
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 0 deletions.
diff --git a/Code/Mantid/Vates/VatesAPI/CMakeLists.txt b/Code/Mantid/Vates/VatesAPI/CMakeLists.txt
@@ -144,6 +144,7 @@ test/vtkMDHistoHex4DFactoryTest.h
 test/vtkMDHistoHexFactoryTest.h
 test/vtkMDHistoLineFactoryTest.h
 test/vtkMDHistoQuadFactoryTest.h
+test/vtkMD0DFactoryTest.h
 test/FieldDataToMetadataTest.h
 test/FilteringUpdateProgressActionTest.h
 test/LoadVTKTest.h

diff --git a/Code/Mantid/Vates/VatesAPI/src/vtkMD0DFactory.cpp b/Code/Mantid/Vates/VatesAPI/src/vtkMD0DFactory.cpp
@@ -1,6 +1,7 @@
 #include "MantidVatesAPI/vtkMD0DFactory.h"
 #include "MantidAPI/IMDWorkspace.h"
 #include "MantidVatesAPI/vtkNullUnstructuredGrid.h"
+#include "MantidVatesAPI/ProgressAction.h"
 
 using namespace Mantid::API;
 
@@ -29,6 +30,7 @@ namespace Mantid
     */
     vtkDataSet* vtkMD0DFactory::create(ProgressAction& progressUpdating) const
     {
+      (void) progressUpdating;
       vtkNullUnstructuredGrid nullGrid;
       vtkUnstructuredGrid *visualDataSet = nullGrid.createNullData();
       return visualDataSet;

diff --git a/Code/Mantid/Vates/VatesAPI/test/vtkMD0DFactoryTest.h b/Code/Mantid/Vates/VatesAPI/test/vtkMD0DFactoryTest.h
@@ -0,0 +1,43 @@
+#ifndef VTK_MD_0D_FACTORY_TEST
+#define VTK_MD_0D_FACTORY_TEST
+#include <cxxtest/TestSuite.h>
+
+#include "MantidVatesAPI/vtkMD0DFactory.h"
+#include "MantidTestHelpers/MDEventsTestHelper.h"
+#include "MantidVatesAPI/UserDefinedThresholdRange.h"
+#include "MantidVatesAPI/NoThresholdRange.h"
+
+#include "MockObjects.h"
+#include <gmock/gmock.h>
+#include <gtest/gtest.h>
+
+#include "MantidVatesAPI/vtkStructuredGrid_Silent.h"
+
+using namespace Mantid;
+using namespace Mantid::VATES;
+using namespace Mantid::API;
+using namespace Mantid::DataObjects;
+using namespace testing;
+
+
+class vtkMD0DFactoryTest : public CxxTest::TestSuite
+{
+public:
+
+  void testCreatesA0DDataSet()
+  {
+    // Arrange
+    FakeProgressAction progressUpdater;
+    vtkMD0DFactory factory(ThresholdRange_scptr(new UserDefinedThresholdRange(0, 1)), "signal");
+
+    vtkDataSet* dataSet = NULL;
+
+    // Assert
+    TSM_ASSERT_THROWS_NOTHING("0D factory should create data set without exceptions", dataSet  = factory.create(progressUpdater));
+    TSM_ASSERT("Should have exactly one point", dataSet->GetNumberOfPoints() == 1);
+    TSM_ASSERT("Should have exactly one cell", dataSet->GetNumberOfCells() == 1);
+  }
+};
+
+
+#endif