From 597f420b31484133cb1e76ba5bf4dff99c773e50 Mon Sep 17 00:00:00 2001
From: Munyoki Kilyungi
Date: Mon, 5 Sep 2022 15:09:58 +0300
Subject: Add extra distinct_values and id attributes to test TraitObject

* wqflask/tests/wqflask/show_trait/test_show_trait.py (TraitObject):
Add self.id and self.distinct_values.
---
 wqflask/tests/wqflask/show_trait/test_show_trait.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'wqflask/tests')

diff --git a/wqflask/tests/wqflask/show_trait/test_show_trait.py b/wqflask/tests/wqflask/show_trait/test_show_trait.py
index b18f8d0a..8dbbb77b 100644
--- a/wqflask/tests/wqflask/show_trait/test_show_trait.py
+++ b/wqflask/tests/wqflask/show_trait/test_show_trait.py
@@ -14,8 +14,11 @@ from wqflask.show_trait.show_trait import get_scales_from_genofile
 
 class TraitObject:
     def __init__(self, obj):
+        self.distinct_values = []
+        self.id = ""
         for key, value in obj.items():
             setattr(self, key, value)
+            self.id += str(value)
 
 
 class TestTraits(unittest.TestCase):
-- 
cgit v1.2.3