def test_single_simple_node(): """ Test basic interface with only a single node. """ node = SimpleNode(id=1) cluster = Cluster((node,)) # hashing lookup assert cluster.location(0) == node assert cluster.location('hello') == node assert cluster.location(sys.maxint) == node # assignment and retrieval cluster['hello'] = 'world' assert cluster['hello'] == 'world' assert len(cluster) == 1 # deletion del cluster['hello'] try: cluster['hello'] assert False except KeyError: assert True assert len(cluster) == 0
def test_multiple_simple_nodes(): """ Test basic interface with multiple nodes. """ nodes = (SimpleNode(id=1), SimpleNode(id=2)) cluster = Cluster(nodes) assert cluster.location(0) == nodes[-1] assert cluster.location(1) == nodes[0] assert cluster.location(2) == nodes[1]