From dd9c48ac81b545b0c6f8ee35ffb3803d4d11cc60 Mon Sep 17 00:00:00 2001
From: Andrei Bondarenko <andrei.i.bondarenko@gmail.com>
Date: Wed, 14 Jul 2021 10:21:49 +0200
Subject: [PATCH] Reverted some changes to state base

---
 state/base.py                       |  4 +--
 state/test/test_create_nodevalue.py | 49 ++++++++++++++++++++++++-----
 2 files changed, 43 insertions(+), 10 deletions(-)

diff --git a/state/base.py b/state/base.py
index 8502f12..242c4b7 100644
--- a/state/base.py
+++ b/state/base.py
@@ -8,9 +8,7 @@ FLOAT = ("Float",)
 STRING = ("String",)
 BOOLEAN = ("Boolean",)
 TYPE = ("Type",)
-NODE = ("Node",)
-EDGE = ("Edge",)
-type_values = (INTEGER, FLOAT, STRING, BOOLEAN, TYPE, NODE, EDGE)
+type_values = (INTEGER, FLOAT, STRING, BOOLEAN, TYPE)
 
 
 Node = str
diff --git a/state/test/test_create_nodevalue.py b/state/test/test_create_nodevalue.py
index 23e12b9..7e3865a 100644
--- a/state/test/test_create_nodevalue.py
+++ b/state/test/test_create_nodevalue.py
@@ -1,5 +1,4 @@
 import pytest
-from state.base import INTEGER, FLOAT, STRING, BOOLEAN, TYPE, NODE, EDGE
 
 
 @pytest.mark.usefixtures("state")
@@ -124,13 +123,49 @@ def test_create_nodevalue_junk(state):
 
 
 @pytest.mark.usefixtures("state")
-def test_create_nodevalue_type(state):
-    for t in {INTEGER, FLOAT, STRING, BOOLEAN, TYPE, NODE, EDGE}:
-        id1 = state.create_nodevalue(t)
-        assert id1 is not None
+def test_create_nodevalue_type_type(state):
+    id1 = state.create_nodevalue(("Type",))
+    assert id1 is not None
+
+    v = state.read_value(id1)
+    assert v == ("Type",)
+
+
+@pytest.mark.usefixtures("state")
+def test_create_nodevalue_integer_type(state):
+    id1 = state.create_nodevalue(("Integer",))
+    assert id1 is not None
+
+    v = state.read_value(id1)
+    assert v == ("Integer",)
+
+
+@pytest.mark.usefixtures("state")
+def test_create_nodevalue_float_type(state):
+    id1 = state.create_nodevalue(("Float",))
+    assert id1 is not None
+
+    v = state.read_value(id1)
+    assert v == ("Float",)
+
+
+@pytest.mark.usefixtures("state")
+def test_create_nodevalue_boolean_type(state):
+    id1 = state.create_nodevalue(("Boolean",))
+    assert id1 is not None
+
+    v = state.read_value(id1)
+    assert v == ("Boolean",)
+
+
+@pytest.mark.usefixtures("state")
+def test_create_nodevalue_string_type(state):
+    id1 = state.create_nodevalue(("String",))
+    assert id1 is not None
+
+    v = state.read_value(id1)
+    assert v == ("String",)
 
-        v = state.read_value(id1)
-        assert v == t
 
 @pytest.mark.usefixtures("state")
 def test_create_nodevalue_invalid_type(state):