tests

Author: ilvalerione

tests/ChatHistoryTest.php

@@ -0,0 +1,63 @@
+<?php
+
+namespace NeuronAI\Tests;
+
+use NeuronAI\AbstractChatHistory;
+use NeuronAI\InMemoryChatHistory;
+use NeuronAI\Messages\UserMessage;
+use PHPUnit\Framework\TestCase;
+
+class ChatHistoryTest extends TestCase
+{
+    /**
+     * Sets up the fixture, for example, open a network connection.
+     * This method is called before a test is executed.
+     *
+     * @throws \Exception
+     */
+    public function setUp(): void
+    {
+    }
+
+    public function testChatHistoryInstance()
+    {
+        $history = new InMemoryChatHistory();
+        $this->assertInstanceOf(AbstractChatHistory::class, $history);
+    }
+
+    public function testChatHistoryAddMessage()
+    {
+        $history = new InMemoryChatHistory();
+        $history->addMessage(new UserMessage('Hello!'));
+        $this->assertCount(1, $history->getMessages());
+        $this->assertEquals(1, $history->count());
+    }
+
+    public function testChatHistoryLastMessage()
+    {
+        $history = new InMemoryChatHistory();
+        $history->addMessage(new UserMessage('Hello!'));
+        $history->addMessage(new UserMessage('Hello2!'));
+        $this->assertEquals('Hello2!', $history->getLastMessage()->getContent());
+    }
+
+    public function testChatHistoryTruncateOldMessages()
+    {
+        $history = new InMemoryChatHistory();
+        $history->addMessage(new UserMessage('Hello!'));
+        $history->addMessage(new UserMessage('Hello2!'));
+        $history->truncate(1);
+        $this->assertEquals(1, $history->count());
+        $this->assertEquals('Hello2!', $history->getLastMessage()->getContent());
+    }
+
+    public function testChatHistoryClear()
+    {
+        $history = new InMemoryChatHistory();
+        $history->addMessage(new UserMessage('Hello!'));
+        $history->addMessage(new UserMessage('Hello2!'));
+        $history->clear();
+        $this->assertCount(0, $history->getMessages());
+        $this->assertEquals(0, $history->count());
+    }
+}

tests/NeuronAITest.php

@@ -3,12 +3,9 @@
 namespace NeuronAI\Tests;
 
 
-use NeuronAI\AbstractChatHistory;
 use NeuronAI\Agent;
-use NeuronAI\InMemoryChatHistory;
-use NeuronAI\Messages\UserMessage;
+use NeuronAI\RAG\RAG;
 use NeuronAI\RAG\VectorStore\MemoryVectorStore;
-use NeuronAI\RAG\VectorStore\VectorStoreInterface;
 use PHPUnit\Framework\TestCase;
 
 class NeuronAITest extends TestCase
@@ -23,37 +20,12 @@ public function setUp(): void
     {
     }
 
-    public function testInstance()
+    public function testAgentInstance()
     {
         $neuron = new Agent();
-        $this->assertInstanceOf(Agent::class, $neuron);
         $this->assertInstanceOf(\SplSubject::class, $neuron);
-    }
-
-    public function testChatHistory()
-    {
-        $history = new InMemoryChatHistory();
-        $this->assertInstanceOf(AbstractChatHistory::class, $history);
-
-        $history->addMessage(new UserMessage('Hello!'));
-        $this->assertCount(1, $history->getMessages());
-        $this->assertEquals(1, $history->count());
-
-        $history->addMessage(new UserMessage('Hello2!'));
-        $this->assertEquals('Hello2!', $history->getLastMessage()->getContent());
 
-        $history->truncate(1);
-        $this->assertEquals(1, $history->count());
-        $this->assertEquals('Hello2!', $history->getLastMessage()->getContent());
-
-        $history->clear();
-        $this->assertCount(0, $history->getMessages());
-        $this->assertEquals(0, $history->count());
-    }
-
-    public function testVectorStore()
-    {
-        $store = new MemoryVectorStore();
-        $this->assertInstanceOf(VectorStoreInterface::class, $store);
+        $neuron = new RAG(new MemoryVectorStore());
+        $this->assertInstanceOf(Agent::class, $neuron);
     }
 }

tests/VectorStoreTes.php

@@ -0,0 +1,20 @@
+<?php
+
+namespace NeuronAI\Tests;
+
+use NeuronAI\RAG\VectorStore\MemoryVectorStore;
+use NeuronAI\RAG\VectorStore\VectorStoreInterface;
+use PHPUnit\Framework\TestCase;
+
+class VectorStoreTes extends TestCase
+{
+    protected function setUp(): void
+    {
+    }
+
+    public function testVectorStoreInstance()
+    {
+        $store = new MemoryVectorStore();
+        $this->assertInstanceOf(VectorStoreInterface::class, $store);
+    }
+}