Python CommBase.get_comm_registry Examples

Example #1

def test_registry():
    r"""Test registry of comm."""
    comm_class = 'CommBase'
    key = 'key1'
    value = None
    assert (not CommBase.is_registered(comm_class, key))
    assert (not CommBase.unregister_comm(comm_class, key))
    assert_equal(CommBase.get_comm_registry(None), {})
    assert_equal(CommBase.get_comm_registry(comm_class), {})
    CommBase.register_comm(comm_class, key, value)
    assert (key in CommBase.get_comm_registry(comm_class))
    assert (CommBase.is_registered(comm_class, key))
    assert (not CommBase.unregister_comm(comm_class, key, dont_close=True))
    CommBase.register_comm(comm_class, key, value)
    assert (not CommBase.unregister_comm(comm_class, key))

Example #2

def test_registry():
    r"""Test registry of comm."""
    from yggdrasil.communication import CommBase
    comm_class = 'CommBase'
    key = 'key1'
    value = None
    assert (not CommBase.is_registered(comm_class, key))
    assert (not CommBase.unregister_comm(comm_class, key))
    assert (CommBase.get_comm_registry(None) == {})
    assert (CommBase.get_comm_registry(comm_class) == {})
    CommBase.register_comm(comm_class, key, value)
    assert (key in CommBase.get_comm_registry(comm_class))
    assert (CommBase.is_registered(comm_class, key))
    assert (not CommBase.unregister_comm(comm_class, key, dont_close=True))
    CommBase.register_comm(comm_class, key, value)
    assert (not CommBase.unregister_comm(comm_class, key))

Example #3

 def test_cleanup_comms(self):
     r"""Test cleanup_comms for comm class."""
     CommBase.cleanup_comms(self.recv_instance.comm_class)
     assert (len(CommBase.get_comm_registry(
         self.recv_instance.comm_class)) == 0)