From b8ccf5fba0872f4bc4fd11b9e960b25350afe533 Mon Sep 17 00:00:00 2001
From: Stefano Borini <sborini@enthought.com>
Date: Tue, 1 Aug 2017 13:18:43 +0100
Subject: [PATCH] Added test for data value

---
 force_bdss/core/data_value.py            |  6 ++++--
 force_bdss/core/tests/__init__.py        |  0
 force_bdss/core/tests/test_data_value.py | 13 +++++++++++++
 3 files changed, 17 insertions(+), 2 deletions(-)
 create mode 100644 force_bdss/core/tests/__init__.py
 create mode 100644 force_bdss/core/tests/test_data_value.py

diff --git a/force_bdss/core/data_value.py b/force_bdss/core/data_value.py
index fc20a49..f360446 100644
--- a/force_bdss/core/data_value.py
+++ b/force_bdss/core/data_value.py
@@ -1,4 +1,4 @@
-from traits.api import HasStrictTraits, Any, String, Int
+from traits.api import HasStrictTraits, Any, String, Int, Enum
 
 
 class DataValue(HasStrictTraits):
@@ -16,9 +16,11 @@ class DataValue(HasStrictTraits):
     #: The value.
     value = Any()
 
+    # The numerical accuracy of the value.
     accuracy = Any()
 
-    quality = Int()
+    #: A flag for the quality of the data.
+    quality = Enum("AVERAGE", "POOR", "GOOD")
 
     def __str__(self):
         return """
diff --git a/force_bdss/core/tests/__init__.py b/force_bdss/core/tests/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/force_bdss/core/tests/test_data_value.py b/force_bdss/core/tests/test_data_value.py
new file mode 100644
index 0000000..c4aa549
--- /dev/null
+++ b/force_bdss/core/tests/test_data_value.py
@@ -0,0 +1,13 @@
+import unittest
+
+from force_bdss.core.data_value import DataValue
+
+
+class TestDataValue(unittest.TestCase):
+    def test_initialization(self):
+        dv = DataValue()
+        self.assertEqual(dv.type, "")
+        self.assertEqual(dv.value, None)
+        self.assertEqual(dv.accuracy, None)
+        self.assertEqual(dv.quality, "AVERAGE")
+
-- 
GitLab