From 8c386fff40a1605c7304b8998adcd1a65263a672 Mon Sep 17 00:00:00 2001
From: Tim Burke <tim.burke@gmail.com>
Date: Fri, 11 May 2018 13:25:17 -0700
Subject: [PATCH] Make the decision between primary/handoff sets more obvious

Change-Id: I419de59df3317d67c594fe768f5696de24148280
---
 test/probe/brain.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/test/probe/brain.py b/test/probe/brain.py
index fd597cf6b3..106749e182 100644
--- a/test/probe/brain.py
+++ b/test/probe/brain.py
@@ -101,10 +101,8 @@ class BrainSplitter(object):
 
         self.part, self.nodes = self.ring.get_nodes(self.account, c, o)
 
-        node_ids = [n['id'] for n in self.nodes]
-        self.node_numbers = [n + 1 for n in node_ids]
-
-        if all(n_id in node_ids for n_id in (0, 1)):
+        self.node_numbers = [n['id'] + 1 for n in self.nodes]
+        if 1 in self.node_numbers and 2 in self.node_numbers:
             self.primary_numbers = (1, 2)
             self.handoff_numbers = (3, 4)
         else: