Fixed the unit test for a more general case. Refs #6862.

mantidproject · Aug 13, 2013 · 0ce0902 · 0ce0902
1 parent fea28f6
commit 0ce0902
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/...ntid/Framework/PythonInterface/test/python/plugins/algorithms/CreateLeBailFitInputTest.py b/...ntid/Framework/PythonInterface/test/python/plugins/algorithms/CreateLeBailFitInputTest.py
@@ -29,10 +29,19 @@ def test_LoadHKLFile(self):
         # Verify some values
         # Profile parameter workspace
         paramws = AnalysisDataService.retrieve("PG3_Bank2_Foo")
-        self.assertEqual(27, paramws.rowCount())
+
+        paramname0 = paramws.cell(0, 0)
+
+        if paramname0.lower() == "bank":
+            numrowgood = 28
+        else:
+            numrowgood = 27
+        print "Parameter name of first line = ", paramname0
+
+        self.assertEqual(numrowgood, paramws.rowCount())
 
         paramnames = []
-        for i in xrange(27):
+        for i in xrange(numrowgood):
             paramname = paramws.cell(i, 0)
             paramnames.append(paramname)
         self.assertEqual(paramnames.count("LatticeConstant"), 1)