<s> import unittest <EOL> import os <EOL> from yaksh import python_code_evaluator <EOL> from yaksh . language_registry import _LanguageRegistry , set_registry , get_registry <EOL> from yaksh . settings import SERVER_TIMEOUT <EOL> class RegistryTestCase ( unittest . TestCase ) : <EOL> def setUp ( self ) : <EOL> set_registry ( ) <EOL> self . registry_object = get_registry ( ) <EOL> self . language_registry = _LanguageRegistry ( ) <EOL> def test_set_register ( self ) : <EOL> class_name = getattr ( python_code_evaluator , '<STR_LIT>' ) <EOL> self . registry_object . register ( "<STR_LIT>" , "<STR_LIT>" ) <EOL> self . assertEquals ( self . registry_object . get_class ( "<STR_LIT>" ) , class_name ) <EOL> def tearDown ( self ) : <EOL> self . registry_object = None <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> unittest . main ( ) </s>
