This commit is contained in:
yumoqing 2025-02-25 21:36:58 +08:00
parent 95dc5fd1f8
commit d75c8ab82f
2 changed files with 44 additions and 0 deletions

21
test/test_prodpricing.py Normal file
View File

@ -0,0 +1,21 @@
from testenv import runtest
from platformbiz.pricing import calc_prod_price
async def test(sor):
prodid = 'a--akNeu1Ia-NOZAJAadf'
spec_config = [
{
"spec_name":"input_tokens",
"count":12832
},
{
"spec_name":"output_tokens",
"count":786323
}
]
price = calc_prod_price(sor, prodid, spec_config)
print(f'{prodid=}, {spec_config=}, {price=}')
if __name__ == '__main__':
runtest(test)

23
test/testenv.py Normal file
View File

@ -0,0 +1,23 @@
import os
import asyncio
from appPublic.jsonConfig import getConfig
from sqlor.dbpools import DBPools
from appbase.init import load_appbase
from accounting.init import load_accounting
from platformbiz.init import load_platformbiz
async def main(asyncfunc):
home = os.environ['HOME']
p = f'{home}/py/sage'
config = getCofig(p, {'workdir':p})
db = DBPools(config.databases)
load_appbase()
load_accounting()
load_platformbiz()
async with db.sqlorContext('sage') as sor:
await asyncfunc(sor)
def runtest(asyncfunc):
asyncio.new_event_loop().run_until_complete(main(asyncfunc))