def test_get_next_node_label(self): con = mock_DatabaseConnector() rep = GeneRepository(con) generated_labels = [] for i in range(100): generated_labels.append( rep.get_next_node_label() ) self.assertListEqual( generated_labels, list(range(100)), "The GeneRepository didn't generate the appropriate node labels." )