BUG: Update test_cmatrices to new interface

AIM-Harvard · Mar 7, 2018 · 62cf084 · 62cf084
1 parent beb7553
commit 62cf084
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_cmatrices.py b/tests/test_cmatrices.py
@@ -33,18 +33,18 @@ def generate_scenarios():
   global testUtils
 
   @parameterized.expand(generate_scenarios(), testcase_func_name=custom_name_func)
-  def test_scenario(self, testCase, featureClassName):
+  def test_scenario(self, test, featureClassName):
     print("")
     global testUtils, featureClasses
 
-    logging.debug('test_scenario: testCase = %s, featureClassName = %s', testCase, featureClassName)
+    logging.debug('test_scenario: testCase = %s, featureClassName = %s', test, featureClassName)
 
     assert cMatsEnabled()
 
-    testUtils.setFeatureClassAndTestCase(featureClassName, testCase)
+    testUtils.setFeatureClassAndTestCase(featureClassName, test)
 
-    testImage = testUtils.getImage()
-    testMask = testUtils.getMask()
+    testImage = testUtils.getImage('original')
+    testMask = testUtils.getMask('original')
 
     featureClass = featureClasses[featureClassName](testImage, testMask, **testUtils.getSettings())