From 44c8dfaa7cf00ea2e3365e40cf899289dad61d7a Mon Sep 17 00:00:00 2001 From: Sudipta Basak Date: Mon, 8 Apr 2024 22:31:55 +0000 Subject: [PATCH] Bug fixes --- ConnectivityTest/MaliciousNodesVsNetDegree/prog.py | 4 ++-- .../MaliciousNodesVsNumberOfNodes/prog.py | 12 ++++++------ 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/ConnectivityTest/MaliciousNodesVsNetDegree/prog.py b/ConnectivityTest/MaliciousNodesVsNetDegree/prog.py index 574d4a6..d6c4bf5 100644 --- a/ConnectivityTest/MaliciousNodesVsNetDegree/prog.py +++ b/ConnectivityTest/MaliciousNodesVsNetDegree/prog.py @@ -37,10 +37,10 @@ def getNodeCountPerColumn(config): colsSelected = random.sample(list(range(1, numberOfCols + 1)), chiC1 if _ < node1Count else chiC2) for col in colsSelected: if col in nodeCountPerColumn.keys(): - nodeCountPerColumn[col][0 if _ < numOfNodes else 1] += 1 + nodeCountPerColumn[col][0 if _ < node1Count else 1] += 1 else: nodeCountPerColumn[col] = [0, 0] - nodeCountPerColumn[col][0 if _ < numOfNodes else 1] = 1 + nodeCountPerColumn[col][0 if _ < node1Count else 1] = 1 return nodeCountPerColumn diff --git a/ConnectivityTest/MaliciousNodesVsNumberOfNodes/prog.py b/ConnectivityTest/MaliciousNodesVsNumberOfNodes/prog.py index 257271d..46b6d7c 100644 --- a/ConnectivityTest/MaliciousNodesVsNumberOfNodes/prog.py +++ b/ConnectivityTest/MaliciousNodesVsNumberOfNodes/prog.py @@ -20,11 +20,11 @@ def plotData(conf): plt.clf() def isGroupRecoverable(group, nodes, chi1, chi2, size, sizeK): - uniqueLines = set() + linesSelected = [] for g in group: - linesSelected = set(random.sample(list(range(1, size + 1)), chi1 if g < nodes[0] else chi2)) - uniqueLines.union(linesSelected) - return len(uniqueLines) >= sizeK + linesSelected += list(random.sample(list(range(1, size + 1)), chi1 if g < nodes[0] else chi2)) + uniqueLines = set(linesSelected) + return len(uniqueLines) >= sizeK, len(uniqueLines) def isGConnected(deg, nodes, mal, config): G = nx.random_regular_graph(deg, sum(nodes)) @@ -54,10 +54,10 @@ def getNodeCountPerColumn(config, numOfNodes): colsSelected = random.sample(list(range(1, numberOfCols + 1)), chiC1 if _ < node1Count else chiC2) for col in colsSelected: if col in nodeCountPerColumn.keys(): - nodeCountPerColumn[col][0 if _ < numOfNodes else 1] += 1 + nodeCountPerColumn[col][0 if _ < node1Count else 1] += 1 else: nodeCountPerColumn[col] = [0, 0] - nodeCountPerColumn[col][0 if _ < numOfNodes else 1] = 1 + nodeCountPerColumn[col][0 if _ < node1Count else 1] = 1 return nodeCountPerColumn