41 lines
2.1 KiB
Python
41 lines
2.1 KiB
Python
"""Run from repository root: python3 -m unittest local.test_pricing -v."""
|
|
import json
|
|
from pathlib import Path
|
|
import subprocess
|
|
import unittest
|
|
from .pricing import price, TIERS
|
|
|
|
class PricingTests(unittest.TestCase):
|
|
def test_every_grade_against_actual_site_javascript(self):
|
|
source = Path('dtf-site.html').read_text()
|
|
tiers = source.split('const FAIXAS=')[1].split('\n};',1)[0]+'\n}'
|
|
calculator = source.split('const cobrar=')[1].split(';',1)[0]
|
|
js = f'const FAIXAS={tiers};const MINIMO_M=1;const cobrar={calculator};'
|
|
js += '''const results=[];for(const mode of Object.keys(FAIXAS))for(let grade=0;grade<=100;grade++)
|
|
for(const metres of [0.01,0.99,1,1.01,1.1,2.75,2.8,2.8000000000000003,19.99,60,12000]){
|
|
const unit=FAIXAS[mode].find(x=>grade>=x[0])[1];
|
|
results.push([mode,String(metres),grade,cobrar(metres),Math.round(unit*100),Math.round(cobrar(metres)*unit*100)]);
|
|
}process.stdout.write(JSON.stringify(results));'''
|
|
cases = json.loads(subprocess.check_output(['node','-e',js],text=True))
|
|
for mode,metres,grade,billed,unit,total in cases:
|
|
with self.subTest(mode=mode,metres=metres,grade=grade):
|
|
result=price(mode,metres,grade)
|
|
self.assertEqual(float(result['billed_metres']),billed)
|
|
self.assertEqual(result['unit_cents'],unit)
|
|
self.assertEqual(result['total_cents'],total)
|
|
|
|
def test_assembly_is_included_at_every_tier(self):
|
|
for grade in range(101):
|
|
self.assertEqual(price('avulsa','1',grade)['total_cents']-price('file','1',grade)['total_cents'],1000)
|
|
self.assertEqual(price('uv','1',grade)['total_cents']-price('uvfile','1',grade)['total_cents'],1400)
|
|
|
|
def test_invalid_inputs(self):
|
|
for value in ('0','-1','NaN','Infinity','12001'):
|
|
with self.assertRaises(ValueError):price('file',value,90)
|
|
for grade in (-1,101,True,89.5):
|
|
with self.assertRaises(ValueError):price('file','1',grade)
|
|
with self.assertRaises(ValueError):price('other','1',90)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|