| 40 | class TestRenamePrimVar( unittest.TestCase ) : |
| 41 | |
| 42 | def test( self ) : |
| 43 | |
| 44 | r = IECore.Reader.create( os.path.join( "test", "IECore", "data", "pdcFiles", "particleShape1.250.pdc" ) ) |
| 45 | r["convertPrimVarNames"].setValue( IECore.BoolData(False) ) |
| 46 | self.assertFalse( r.parameters()["convertPrimVarNames"].getTypedValue() ) |
| 47 | p = r.read() |
| 48 | |
| 49 | self.assertTrue( "position" in p ) |
| 50 | self.assertTrue( "particleId" in p ) |
| 51 | |
| 52 | newNames = [ "position P", "particleId id" ] |
| 53 | |
| 54 | o = IECoreScene.RenamePrimitiveVariables() |
| 55 | o["input"].setValue( p ) |
| 56 | o["names"].setValue( IECore.StringVectorData( newNames ) ) |
| 57 | |
| 58 | pp = o() |
| 59 | self.assertEqual( len( pp ), len( p ) ) |
| 60 | for n in newNames : |
| 61 | ns = n.split() |
| 62 | self.assertTrue( ns[1] in pp ) |
| 63 | self.assertTrue( not ns[0] in pp ) |
| 64 | |
| 65 | if __name__ == "__main__": |
| 66 | unittest.main() |