Skip to content
Snippets Groups Projects
Normal view Permalink
BinaryOpHelperTest.h 8.15 KiB
Newer Older
Nick Draper's avatar
re #59
Nick Draper committed
1 2 3 4 5 6 
#ifndef BINARYOPHELPERTEST_H_
#define BINARYOPHELPERTEST_H_

#include <cxxtest/TestSuite.h>
#include <cmath>
Nick Draper's avatar
re #59  
Nick Draper committed
7 
#include "WorkspaceCreationHelper.hh"
Nick Draper's avatar
re #59
Nick Draper committed
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 
#include "MantidAlgorithms/BinaryOpHelper.h"
#include "MantidAPI/AnalysisDataService.h"
#include "MantidAPI/WorkspaceFactory.h"
#include "MantidAPI/TripleRef.h" 
#include "MantidAPI/TripleIterator.h" 
#include "MantidDataObjects/Workspace2D.h"
#include "MantidDataObjects/Workspace1D.h"

using namespace Mantid::API;
using namespace Mantid::Kernel;
using namespace Mantid::Algorithms;
using namespace Mantid::DataObjects;

class BinaryOpHelperTest : public CxxTest::TestSuite
{
public:

  void testcheckSizeCompatability1D1D()
  {
    int sizex = 10;
    // Register the workspace in the data service
Nick Draper's avatar
re #68  
Nick Draper committed
29 30 31 32 33 34 35 
    Workspace1D_sptr work_in1 = WorkspaceCreationHelper::Create1DWorkspaceFib(sizex);
    Workspace1D_sptr work_in2 = WorkspaceCreationHelper::Create1DWorkspaceFib(20);
    Workspace1D_sptr work_in3 = WorkspaceCreationHelper::Create1DWorkspaceFib(10);
    Workspace1D_sptr work_in4 = WorkspaceCreationHelper::Create1DWorkspaceFib(5);
    Workspace1D_sptr work_in5 = WorkspaceCreationHelper::Create1DWorkspaceFib(3);
    Workspace1D_sptr work_in6 = WorkspaceCreationHelper::Create1DWorkspaceFib(1);
    Workspace1D_sptr work_in7 = WorkspaceCreationHelper::Create1DWorkspaceFib(0);
Nick Draper's avatar
re #59
Nick Draper committed
36 37 38 39 40 41 
    BinaryOpHelper helper;
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in2));
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in3));
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in4));
    TS_ASSERT(!helper.checkSizeCompatability(work_in1,work_in5));
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in6));
Nick Draper's avatar
re #68  
Nick Draper committed
42 
    TS_ASSERT(!helper.checkSizeCompatability(work_in1,work_in7));
Nick Draper's avatar
re #59
Nick Draper committed
43 44 45 46 47 
  }

  void testcheckSizeCompatability2D1D()
  {
    // Register the workspace in the data service
Nick Draper's avatar
re #68  
Nick Draper committed
48 49 50 51 52 53 54 
    Workspace2D_sptr work_in1 = WorkspaceCreationHelper::Create2DWorkspace(10,10);
    Workspace1D_sptr work_in2 = WorkspaceCreationHelper::Create1DWorkspaceFib(20);
    Workspace1D_sptr work_in3 = WorkspaceCreationHelper::Create1DWorkspaceFib(10);
    Workspace1D_sptr work_in4 = WorkspaceCreationHelper::Create1DWorkspaceFib(5);
    Workspace1D_sptr work_in5 = WorkspaceCreationHelper::Create1DWorkspaceFib(3);
    Workspace1D_sptr work_in6 = WorkspaceCreationHelper::Create1DWorkspaceFib(1);
    Workspace1D_sptr work_in7 = WorkspaceCreationHelper::Create1DWorkspaceFib(0);
Nick Draper's avatar
re #59
Nick Draper committed
55 56 57 58 59 60 
    BinaryOpHelper helper;
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in2));
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in3));
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in4));
    TS_ASSERT(!helper.checkSizeCompatability(work_in1,work_in5));
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in6));
Nick Draper's avatar
re #68  
Nick Draper committed
61 
    TS_ASSERT(!helper.checkSizeCompatability(work_in1,work_in7));
Nick Draper's avatar
re #59
Nick Draper committed
62 63 64 65 66 67 
  }

  
  void testcheckSizeCompatability2D2D()
  {
    // Register the workspace in the data service
Nick Draper's avatar
re #68  
Nick Draper committed
68 69 70 71 72 73 74 
    Workspace2D_sptr work_in1 = WorkspaceCreationHelper::Create2DWorkspace(10,10);
    Workspace2D_sptr work_in2 = WorkspaceCreationHelper::Create2DWorkspace(20,10);
    Workspace2D_sptr work_in3 = WorkspaceCreationHelper::Create2DWorkspace(10,10);
    Workspace2D_sptr work_in4 = WorkspaceCreationHelper::Create2DWorkspace(5,5);
    Workspace2D_sptr work_in5 = WorkspaceCreationHelper::Create2DWorkspace(3,3);
    Workspace2D_sptr work_in6 = WorkspaceCreationHelper::Create2DWorkspace(1,100);
    Workspace2D_sptr work_in7 = WorkspaceCreationHelper::Create2DWorkspace(0,0);
Nick Draper's avatar
re #59
Nick Draper committed
75 76 77 78 79 80 
    BinaryOpHelper helper;
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in2));
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in3));
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in4));
    TS_ASSERT(!helper.checkSizeCompatability(work_in1,work_in5));
    TS_ASSERT(helper.checkSizeCompatability(work_in1,work_in6));
Nick Draper's avatar
re #68  
Nick Draper committed
81 
    TS_ASSERT(!helper.checkSizeCompatability(work_in1,work_in7));
Nick Draper's avatar
re #59
Nick Draper committed
82 83 84 85 86 
  }

  void testcreateOutputWorkspace1D1D()
  {
    // Register the workspace in the data service
Nick Draper's avatar
re #68  
Nick Draper committed
87 88 89 90 91 92 93 
    Workspace1D_sptr work_in1 = WorkspaceCreationHelper::Create1DWorkspaceFib(10);
    Workspace1D_sptr work_in2 = WorkspaceCreationHelper::Create1DWorkspaceFib(20);
    Workspace1D_sptr work_in3 = WorkspaceCreationHelper::Create1DWorkspaceFib(10);
    Workspace1D_sptr work_in4 = WorkspaceCreationHelper::Create1DWorkspaceFib(5);
    Workspace1D_sptr work_in5 = WorkspaceCreationHelper::Create1DWorkspaceFib(3);
    Workspace1D_sptr work_in6 = WorkspaceCreationHelper::Create1DWorkspaceFib(1);
    Workspace1D_sptr work_in7 = WorkspaceCreationHelper::Create1DWorkspaceFib(0);
Nick Draper's avatar
re #59
Nick Draper committed
94 95 96 97 98 99 
    BinaryOpHelper helper;
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in2),work_in1,work_in2);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in3),work_in1,work_in3);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in4),work_in1,work_in4);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in5),work_in1,work_in5);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in6),work_in1,work_in6);
Nick Draper's avatar
re #68  
Nick Draper committed
100 
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in7),work_in1,work_in7);
Nick Draper's avatar
re #59
Nick Draper committed
101 102 103 104 105 
  }
  
  void testcreateOutputWorkspace2D1D()
  {
    // Register the workspace in the data service
Nick Draper's avatar
re #68  
Nick Draper committed
106 107 108 109 110 111 112 
    Workspace2D_sptr work_in1 = WorkspaceCreationHelper::Create2DWorkspace(5,2);
    Workspace1D_sptr work_in2 = WorkspaceCreationHelper::Create1DWorkspaceFib(20);
    Workspace1D_sptr work_in3 = WorkspaceCreationHelper::Create1DWorkspaceFib(10);
    Workspace1D_sptr work_in4 = WorkspaceCreationHelper::Create1DWorkspaceFib(5);
    Workspace1D_sptr work_in5 = WorkspaceCreationHelper::Create1DWorkspaceFib(3);
    Workspace1D_sptr work_in6 = WorkspaceCreationHelper::Create1DWorkspaceFib(1);
    Workspace1D_sptr work_in7 = WorkspaceCreationHelper::Create1DWorkspaceFib(0);
Nick Draper's avatar
re #59
Nick Draper committed
113 114 115 116 117 118 
    BinaryOpHelper helper;
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in2),work_in1,work_in2);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in3),work_in1,work_in3);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in4),work_in1,work_in4);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in5),work_in1,work_in5);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in6),work_in1,work_in6);
Nick Draper's avatar
re #68  
Nick Draper committed
119 
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in7),work_in1,work_in7);
Nick Draper's avatar
re #59
Nick Draper committed
120 121 122 123 124 125 
  }

 
  void testcreateOutputWorkspace2D2D()
  {
    // Register the workspace in the data service
Nick Draper's avatar
re #68  
Nick Draper committed
126 127 128 129 130 131 132 
    Workspace2D_sptr work_in1 = WorkspaceCreationHelper::Create2DWorkspace(10,10);
    Workspace2D_sptr work_in2 = WorkspaceCreationHelper::Create2DWorkspace(20,10);
    Workspace2D_sptr work_in3 = WorkspaceCreationHelper::Create2DWorkspace(10,10);
    Workspace2D_sptr work_in4 = WorkspaceCreationHelper::Create2DWorkspace(5,5);
    Workspace2D_sptr work_in5 = WorkspaceCreationHelper::Create2DWorkspace(3,3);
    Workspace2D_sptr work_in6 = WorkspaceCreationHelper::Create2DWorkspace(1,100);
    Workspace2D_sptr work_in7 = WorkspaceCreationHelper::Create2DWorkspace(0,0);
Nick Draper's avatar
re #59
Nick Draper committed
133 134 135 136 137 138 
    BinaryOpHelper helper;
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in2),work_in1,work_in2);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in3),work_in1,work_in3);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in4),work_in1,work_in4);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in5),work_in1,work_in5);
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in6),work_in1,work_in6);
Nick Draper's avatar
re #68  
Nick Draper committed
139 
    checkOutputWorkspace(helper.createOutputWorkspace(work_in1,work_in7),work_in1,work_in7);
Nick Draper's avatar
re #59
Nick Draper committed
140 141 142 
  }
Nick Draper's avatar
re #68  
Nick Draper committed
143 
  void checkOutputWorkspace(Workspace_sptr ws, Workspace_sptr wsIn1,Workspace_sptr wsIn2 ) const
Nick Draper's avatar
re #59
Nick Draper committed
144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 
  {
    int targetsize = (wsIn1->size()>wsIn2->size())?wsIn1->size():wsIn2->size();
    TS_ASSERT_EQUALS(ws->size(),targetsize);
    //check they arre all 0
    for(triple_iterator<Workspace> ti(*ws); ti != ti.end(); ++ti)
    {
      TS_ASSERT_THROWS_NOTHING
      (
        TripleRef<double&> tr = *ti;
        TS_ASSERT_DELTA(tr[0],0,0.0001);
        TS_ASSERT_DELTA(tr[1],0,0.0001);
        TS_ASSERT_DELTA(tr[2],0,0.0001);
      )
    }
  }
  
};

#endif /*BINARYOPHELPERTEST_H_*/