Merge pull request #25 from Arend-Jan/24-adding-better-code-coverage-for-clientrs

adding more test

Author: Arend-Jan

src/client.rs

@@ -237,4 +237,32 @@ mod tests {
         let result = client.chat(input).await;
         assert!(result.is_err());
     }
+
+    #[test]
+    fn test_usage_struct() {
+        let usage = Usage {
+            prompt_tokens: 10,
+            completion_tokens: 20,
+            total_tokens: 30,
+        };
+
+        assert_eq!(usage.prompt_tokens, 10);
+        assert_eq!(usage.completion_tokens, 20);
+        assert_eq!(usage.total_tokens, 30);
+    }
+
+    #[test]
+    fn test_choice_struct() {
+        let choice = Choice {
+            message: Message {
+                role: Role::Assistant,
+                content: "Sample response".to_string(),
+            },
+            finish_reason: "stop".to_string(),
+        };
+
+        assert_eq!(choice.message.role, Role::Assistant);
+        assert_eq!(choice.message.content, "Sample response");
+        assert_eq!(choice.finish_reason, "stop");
+    }
 }

src/models.rs

@@ -149,4 +149,29 @@ mod tests {
         let deserialized_model: Model = serde_json::from_str(model_json).unwrap();
         assert_eq!(deserialized_model, Model::Gpt_4);
     }
+
+    #[test]
+    fn test_logit_bias_struct() {
+        let mut biases = HashMap::new();
+        biases.insert(42, 2.5);
+        biases.insert(123, -1.3);
+
+        let logit_bias = LogitBias { biases };
+
+        assert_eq!(
+            logit_bias.biases.get(&42),
+            Some(&2.5),
+            "Bias for token 42 should be 2.5"
+        );
+        assert_eq!(
+            logit_bias.biases.get(&123),
+            Some(&-1.3),
+            "Bias for token 123 should be -1.3"
+        );
+        assert_eq!(
+            logit_bias.biases.get(&999),
+            None,
+            "Bias for token 999 should not be set"
+        );
+    }
 }