11# -*- coding: utf-8 -*-
22
3+ import base64
34import json
45import unittest
5- import msgpack
6- import base64
76from datetime import datetime
87from unittest import mock
98
9+ import msgpack
1010from dapr .ext .langgraph .dapr_checkpointer import DaprCheckpointer
1111from langgraph .checkpoint .base import Checkpoint
1212
@@ -69,13 +69,15 @@ def test_put_saves_checkpoint_and_registry(self, mock_client_cls):
6969 first_call_kwargs = first_call .kwargs
7070 assert first_call_kwargs ['store_name' ] == 'statestore'
7171 assert first_call_kwargs ['key' ] == 'checkpoint:t1::cp1'
72- unpacked = msgpack .unpackb (first_call_kwargs ['value' ]) # We're packing bytes
72+ unpacked = msgpack .unpackb (first_call_kwargs ['value' ]) # We're packing bytes
7373 saved_payload = {}
7474 for k , v in unpacked .items ():
7575 k = k .decode () if isinstance (k , bytes ) else k
76- if k == 'checkpoint' or k == 'metadata' : # Need to convert b'' on checkpoint/metadata dict key/values
76+ if (
77+ k == 'checkpoint' or k == 'metadata'
78+ ): # Need to convert b'' on checkpoint/metadata dict key/values
7779 if k == 'metadata' :
78- v = msgpack .unpackb (v ) # Metadata value is packed
80+ v = msgpack .unpackb (v ) # Metadata value is packed
7981 val = {}
8082 for sk , sv in v .items ():
8183 sk = sk .decode () if isinstance (sk , bytes ) else sk
@@ -89,7 +91,9 @@ def test_put_saves_checkpoint_and_registry(self, mock_client_cls):
8991 second_call = mock_client .save_state .call_args_list [1 ]
9092 second_call_kwargs = second_call .kwargs
9193 assert second_call_kwargs ['store_name' ] == 'statestore'
92- assert second_call_kwargs ['value' ] == 'checkpoint:t1::cp1' # Here we're testing if the last checkpoint is the first_call above
94+ assert (
95+ second_call_kwargs ['value' ] == 'checkpoint:t1::cp1'
96+ ) # Here we're testing if the last checkpoint is the first_call above
9397
9498 def test_put_writes_updates_channel_values (self , mock_client_cls ):
9599 mock_client = mock_client_cls .return_value
0 commit comments