mirror of
https://github.com/LostRuins/koboldcpp.git
synced 2026-09-06 04:51:17 +02:00
yhz5613813
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
import math
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
|
|
|
import koboldcpp
|
|
|
|
|
|
class NumericParsingTests(unittest.TestCase):
|
|
def test_valid_numeric_values_keep_existing_behavior(self):
|
|
self.assertEqual(koboldcpp.tryparseint("12", 17), 12)
|
|
self.assertEqual(koboldcpp.tryparseint("true", 17), 1)
|
|
self.assertEqual(koboldcpp.tryparseint("false", 17), 0)
|
|
self.assertEqual(koboldcpp.tryparsefloat("0.75", 0.25), 0.75)
|
|
|
|
def test_tryparseint_returns_fallback_for_incompatible_json_types(self):
|
|
for value in ([], {}, [1], {"value": 1}):
|
|
with self.subTest(value=value):
|
|
self.assertEqual(koboldcpp.tryparseint(value, 17), 17)
|
|
|
|
def test_tryparseint_returns_fallback_for_overflow(self):
|
|
self.assertEqual(koboldcpp.tryparseint(math.inf, 17), 17)
|
|
|
|
def test_tryparsefloat_returns_fallback_for_incompatible_json_types(self):
|
|
for value in ([], {}, [1], {"value": 1}):
|
|
with self.subTest(value=value):
|
|
self.assertEqual(koboldcpp.tryparsefloat(value, 0.25), 0.25)
|
|
|
|
def test_tryparsefloat_returns_fallback_for_non_finite_values(self):
|
|
for value in (math.inf, -math.inf, math.nan, "inf", "nan", 10**10000):
|
|
with self.subTest(value=value):
|
|
self.assertEqual(koboldcpp.tryparsefloat(value, 0.25), 0.25)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|