File size: 729 Bytes
c768949
 
 
a091159
 
c768949
 
 
a091159
 
 
c768949
 
a091159
 
c768949
 
 
a091159
 
c768949
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from pytest import raises
from megabots import memory
from megabots.memory import ConversationBuffer, ConversationBufferWindow


def test_memory_conversation_buffer():
    mem = memory(name="conversation-buffer")
    assert isinstance(mem, ConversationBuffer)


def test_memory_conversation_buffer_window():
    mem = memory(name="conversation-buffer-window", k=10)
    assert isinstance(mem, ConversationBufferWindow)


def test_memory_unsupported_name():
    with raises(ValueError, match=r"Memory invalid-name is not supported."):
        memory(name="invalid-name")


def test_memory_no_name():
    with raises(
        RuntimeError, match=r"Impossible to instantiate memory without a name."
    ):
        memory(name=None)