Method testResize

test/IECore/ObjectMatrixTest.py:187–216

( self )

		self.assertEqual( repr( m ), "IECore.ObjectMatrix( [ [ IECore.IntData( 42 ), IECore.StringData( '123' ) ], [ IECore.FloatData( 0.1 ), IECore.CompoundData() ], [ None, IECore.StringVectorData( [ 'a' ] ) ] ] )" )

	def testResize( self ) :

		m1 = IECore.ObjectMatrix( 3, 3 )
		for row in range( 3 ) :
			for column in range( 3 ) :
				m1[row, column] = IECore.IntData( row * 3 + column )

		m2 = m1.copy()
		self.assertEqual( m1, m2 )

		# Enlarge m2. Its values should match their equivalent coordinate in m1.
		m2.resize( 5, 4 )
		self.assertNotEqual( m1, m2 )

		for row in range( m2.numRows() ) :
			for column in range( m2.numColumns() ) :
				if row < m1.numRows() and column < m1.numColumns() :
					self.assertEqual( m1[row, column], m2[row, column] )
				else :
					self.assertIsNone( m2[row, column] )

		# Shrink m2 back to the same size as m1. Both should match.
		m2.resize( 3, 3 )
		self.assertEqual( m1, m2 )

		# Shrink m2 smaller than m1, remaining values should still match their equivalent coordinate in m1.
		m2.resize( 2, 2 )
		for row in range( m2.numRows() ) :
			for column in range( m2.numRows() ) :
				self.assertEqual( m1[row, column], m2[row, column] )

if __name__ == "__main__":
	unittest.main()