import java.util.Iterator;
import java.util.Map;
+import junit.extensions.PA;
+
import org.testng.Assert;
import org.testng.FileAssert;
import org.testng.annotations.AfterTest;
// Can't do Assert.assertEquals(strucMapping.getMapping(), expectedMapping);
// because this fails in our version of TestNG
- Assert.assertEquals(strucMapping.getMapping().size(),
+ Map<Integer, int[]> mapping = (Map<Integer, int[]>) PA.getValue(
+ strucMapping, "mapping");
+ Assert.assertEquals(
+ mapping.size(),
expectedMapping.size());
Iterator<Map.Entry<Integer, int[]>> it = expectedMapping.entrySet()
.iterator();
while (it.hasNext())
{
Map.Entry<Integer, int[]> pair = it.next();
- Assert.assertTrue(strucMapping.getMapping()
+ Assert.assertTrue(mapping
.containsKey(pair.getKey()));
- Assert.assertEquals(strucMapping.getMapping().get(pair.getKey()),
+ Assert.assertEquals(mapping.get(pair.getKey()),
pair.getValue());
}
}