fixup! Simplify ContextLocalState API

Author: imjoehaines

tests/test_context.py

@@ -91,6 +91,12 @@ def test_state_is_stored_separately_per_async_context():
     assert client1.feature_flags == []
     assert client2.feature_flags == []
 
+    client1.add_feature_flag('a')
+    client2.add_feature_flag('b')
+
+    assert client1.feature_flags == [FeatureFlag('a')]
+    assert client2.feature_flags == [FeatureFlag('b')]
+
     async def mutate_state(id):
         create_new_context()
 
@@ -135,5 +141,5 @@ async def test():
     finally:
         loop.close()
 
-    assert client1.feature_flags == []
-    assert client2.feature_flags == []
+    assert client1.feature_flags == [FeatureFlag('a')]
+    assert client2.feature_flags == [FeatureFlag('b')]