From 7a1bf04e4edb1fbab958a71f49bf68ba8ae687ce Mon Sep 17 00:00:00 2001
From: Paul McCarthy <pauldmccarthy@gmail.com>
Date: Thu, 3 May 2018 11:31:26 +0100
Subject: [PATCH] TEST: Fixed broken assumptions (see previous commit)

---
 tests/test_atlasq_ohi.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_atlasq_ohi.py b/tests/test_atlasq_ohi.py
index 96c61572f..c4a0938d5 100644
--- a/tests/test_atlasq_ohi.py
+++ b/tests/test_atlasq_ohi.py
@@ -155,8 +155,8 @@ def test_mask(seed):
         exp           = []
 
         for lbl, prop in zip(labels, props):
-            exp.append('{}:{:0.4f}'.format(desc.labels[int(lbl)].name,
-                                           prop))
+            name = desc.find(value=int(lbl)).name
+            exp.append('{}:{:0.4f}'.format(name, prop))
 
         return '\n'.join(exp)
 
-- 
GitLab