Make NullContext an alias of ContextBase

Author: github-actions[bot]

ccflow/tests/test_callable.py

@@ -11,7 +11,6 @@
     CallableModelGenericType,
     ContextBase,
     ContextType,
-    DateContext,
     Flow,
     GenericResult,
     GraphDepList,
@@ -33,6 +32,10 @@ class MyExtendedContext(MyContext):
     c: bool
 
 
+class MyOtherContext(ContextBase):
+    a: int
+
+
 class ListContext(ContextBase):
     ll: List[str] = []
 
@@ -51,12 +54,6 @@ def __call__(self, context: MyContext) -> MyResult:
         return MyResult(x=self.i, y=context.a)
 
 
-class MyCallableNullContext(CallableModel):
-    @Flow.call
-    def __call__(self, context: NullContext) -> MyResult:
-        return MyResult(x=1, y="default")
-
-
 class MyCallableOptionalContext(CallableModel):
     @Flow.call
     def __call__(self, context: Optional[MyContext] = None) -> MyResult:
@@ -159,7 +156,7 @@ class BadModelMismatchedContextAndCall(CallableModel):
 
     @property
     def context_type(self):
-        return DateContext
+        return MyOtherContext
 
     @property
     def result_type(self):
@@ -170,7 +167,7 @@ def __call__(self, context: MyContext) -> MyResult:
         return context
 
 
-class BadModelGenericMismatchedContextAndCall(CallableModelGenericType[DateContext, MyResult]):
+class BadModelGenericMismatchedContextAndCall(CallableModelGenericType[MyOtherContext, MyResult]):
     """Model with mismatched context_type and __call__ annotation"""
 
     @Flow.call
@@ -412,11 +409,6 @@ def test_signature(self):
         self.assertRaises(TypeError, m, context, a="foo")
         self.assertRaises(TypeError, m, context=context, a="foo")
 
-    def test_nullable_context(self):
-        m = MyCallableNullContext()
-        context = MyContext(a="foo")
-        self.assertEqual(m(context=context), m(None))
-
     def test_signature_optional_context(self):
         m = MyCallableOptionalContext()
         context = MyContext(a="foo")
@@ -472,7 +464,7 @@ def test_types(self):
         error = "__call__ method must take a single argument, named 'context'"
         self.assertRaisesRegex(ValueError, error, BadModelDoubleContextArg)
 
-        error = "The context_type <class 'ccflow.context.DateContext'> must match the type of the context accepted by __call__ <class 'ccflow.tests.test_callable.MyContext'>"
+        error = "The context_type <class 'ccflow.tests.test_callable.MyOtherContext'> must match the type of the context accepted by __call__ <class 'ccflow.tests.test_callable.MyContext'>"
         self.assertRaisesRegex(ValueError, error, BadModelMismatchedContextAndCall)
 
         error = "The result_type <class 'ccflow.result.generic.GenericResult'> must match the return type of __call__ <class 'ccflow.tests.test_callable.MyResult'>"
@@ -654,7 +646,7 @@ def __call__(self, context: NullContext) -> GenericResult[float]:
             MyCallable()
 
     def test_types_generic(self):
-        error = "Context type annotation <class 'ccflow.tests.test_callable.MyContext'> on __call__ does not match context_type <class 'ccflow.context.DateContext'> defined by CallableModelGenericType"
+        error = "Context type annotation <class 'ccflow.tests.test_callable.MyContext'> on __call__ does not match context_type <class 'ccflow.tests.test_callable.MyOtherContext'> defined by CallableModelGenericType"
         self.assertRaisesRegex(TypeError, error, BadModelGenericMismatchedContextAndCall)
 
         error = "Return type annotation <class 'ccflow.tests.test_callable.MyResult'> on __call__ does not match result_type <class 'ccflow.result.generic.GenericResult'> defined by CallableModelGenericType"

ccflow/tests/test_context.py

@@ -30,6 +30,11 @@
 from ccflow.result import GenericResult
 
 
+class MyDefaultContext(ContextBase):
+    b: float = 3.14
+    c: bool = False
+
+
 class TestContexts(TestCase):
     def test_null_context(self):
         n1 = NullContext()
@@ -38,11 +43,25 @@ def test_null_context(self):
         self.assertEqual(hash(n1), hash(n2))
 
     def test_null_context_validation(self):
+        # Context creation is based on two main assumptions:
+        # 1. If there is enough information to create a context, it should be created.
+        # 2. Since NullContext has no required fields, it can be created from None,
+        #    empty containers ({} or []), or any other context.
         self.assertEqual(NullContext.model_validate([]), NullContext())
         self.assertEqual(NullContext.model_validate({}), NullContext())
         self.assertEqual(NullContext.model_validate(None), NullContext())
+        self.assertIsInstance(NullContext.model_validate(DateContext(date="0d")), NullContext)
         self.assertRaises(ValueError, NullContext.model_validate, [True])
 
+    def test_context_with_defaults(self):
+        # Contexts may define default values. Extending the assumptions above:
+        # Any context inherits the behavior from NullContext, and can be
+        # created as long as all required fields (if any) are satisfied.
+        self.assertEqual(TypeAdapter(MyDefaultContext).validate_python(None), MyDefaultContext(b=3.14, c=False))
+        self.assertEqual(TypeAdapter(MyDefaultContext).validate_python({}), MyDefaultContext(b=3.14, c=False))
+        self.assertEqual(TypeAdapter(MyDefaultContext).validate_python([]), MyDefaultContext(b=3.14, c=False))
+        self.assertEqual(TypeAdapter(MyDefaultContext).validate_python({"b": 10.0}), MyDefaultContext(b=10.0, c=False))
+
     def test_date_validation(self):
         c = DateContext(date=date.today())
         self.assertEqual(DateContext(date=str(date.today())), c)