text: ##########################################################################
  #
  #  Copyright (c) 2010, Image Engine Design Inc. All rights reserved.
  #
  #  Redistribution and use in source and binary forms, with or without
  #  modification, are permitted provided that the following conditions are
  #  met:
  #
  #     * Redistributions of source code must retain the above copyright
  #       notice, this list of conditions and the following disclaimer.
  #
  #     * Redistributions in binary form must reproduce the above copyright
  #       notice, this list of conditions and the following disclaimer in the
  #       documentation and/or other materials provided with the distribution.
  #
  #     * Neither the name of Image Engine Design nor the names of any
  #       other contributors to this software may be used to endorse or
  #       promote products derived from this software without specific prior
  #       written permission.
  #
  #  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
  #  IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  #  THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  #  PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  #  CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  #  EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  #  PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  #  PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  #  LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  #  NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  #  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  #
  ##########################################################################
  
  from __future__ import with_statement
  
  import hou
  import IECoreHoudini
  import unittest
  
  class TestActiveTake( IECoreHoudini.TestCase ) :
  
  	# testing a take can be set with the context
  	def testSetTake( self ) :
  
  		take = "take1"
  		hou.hscript( "takeadd %s" % take )
  
  		self.assertTrue( take in IECoreHoudini.ActiveTake.ls() )
  		self.assertNotEqual( take, IECoreHoudini.ActiveTake.name() )
  
  		with IECoreHoudini.ActiveTake( take ) :
  
  			self.assertEqual( take, IECoreHoudini.ActiveTake.name() )
  
  		self.assertNotEqual( take, IECoreHoudini.ActiveTake.name() )
  
  	# testing the current take can still be used
  	def testSetCurrentTake( self ) :
  
  		take = IECoreHoudini.ActiveTake.name()
  
  		self.assertTrue( take in IECoreHoudini.ActiveTake.ls() )
  		self.assertEqual( take, IECoreHoudini.ActiveTake.name() )
  
  		with IECoreHoudini.ActiveTake( take ) :
  
  			self.assertEqual( take, IECoreHoudini.ActiveTake.name() )
  
  		self.assertEqual( take, IECoreHoudini.ActiveTake.name() )
  
  	# testing a non-existant take won't break anything
  	def testFakeTake( self ) :
  
  		take = "fake"
  		origTake = IECoreHoudini.ActiveTake.name()
  
  		self.assertTrue( take not in IECoreHoudini.ActiveTake.ls() )
  		self.assertNotEqual( take, IECoreHoudini.ActiveTake.name() )
  
  		with IECoreHoudini.ActiveTake( take ) :
  
  			self.assertEqual( origTake, IECoreHoudini.ActiveTake.name() )
  			self.assertNotEqual( take, IECoreHoudini.ActiveTake.name() )
  
  		self.assertEqual( origTake, IECoreHoudini.ActiveTake.name() )
  		self.assertNotEqual( take, IECoreHoudini.ActiveTake.name() )
  
  	# testing deleting a take in the context
  	def testDeleteTake( self ) :
  
  		take = "take1"
  		hou.hscript( "takeadd %s" % take )
  
  		self.assertTrue( take in IECoreHoudini.ActiveTake.ls() )
  		self.assertNotEqual( take, IECoreHoudini.ActiveTake.name() )
  
  		with IECoreHoudini.ActiveTake( take ) :
  
  			self.assertEqual( take, IECoreHoudini.ActiveTake.name() )
  			hou.hscript( "takerm %s" % take )
  			self.assertNotEqual( take, IECoreHoudini.ActiveTake.name() )
  
  		self.assertNotEqual( take, IECoreHoudini.ActiveTake.name() )
  
  if __name__ == "__main__":
      unittest.main()
  