76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
"""
|
|
Verification test suite for Production Backend Engine
|
|
"""
|
|
import sys
|
|
import time
|
|
import os
|
|
|
|
# Add backend directory to sys.path
|
|
sys.path.insert(0, os.path.dirname(__file__))
|
|
|
|
import engine
|
|
|
|
def test_initial_state():
|
|
state = engine.get_initial_state()
|
|
assert state['cash'] == 30.0
|
|
assert 'wheat' in state['products']
|
|
assert state['products']['wheat']['unlocked'] is True
|
|
assert state['products']['wheat']['lines'] == 1
|
|
assert state['products']['flour']['unlocked'] is False
|
|
print("PASS: test_initial_state")
|
|
|
|
def test_raw_production_and_logistics():
|
|
state = engine.get_initial_state()
|
|
# Fast forward 10 seconds
|
|
state['lastTick'] = time.time() - 10.0
|
|
updated = engine.process_offline_and_ticks(state)
|
|
|
|
# Wheat has cycle time 1.0s, output 1, 1 line, plus logistics selling
|
|
assert updated['products']['wheat']['totalProduced'] > 0
|
|
assert updated['cash'] > 30.0
|
|
print(f"PASS: test_raw_production_and_logistics (Produced: {updated['products']['wheat']['totalProduced']}, Cash: {updated['cash']})")
|
|
|
|
def test_chained_dependency_and_bottleneck():
|
|
state = engine.get_initial_state()
|
|
# Turn off wheat lines to simulate 0 upstream production
|
|
state['products']['wheat']['lines'] = 0
|
|
state['products']['wheat']['stock'] = 0.0
|
|
|
|
# Unlock flour and give it 1 line
|
|
state['products']['flour']['unlocked'] = True
|
|
state['products']['flour']['lines'] = 1
|
|
|
|
# 1. Process with no wheat: flour should bottleneck
|
|
state['lastTick'] = time.time() - 5.0
|
|
state = engine.process_offline_and_ticks(state)
|
|
assert state['products']['flour']['stock'] == 0.0
|
|
assert state['products']['flour']['isBottlenecked'] is True
|
|
|
|
# 2. Give 10 wheat: flour should consume wheat and produce flour
|
|
state['products']['wheat']['stock'] = 10.0
|
|
state['lastTick'] = time.time() - 4.0
|
|
state = engine.process_offline_and_ticks(state)
|
|
assert state['products']['flour']['totalProduced'] > 0
|
|
assert state['products']['wheat']['stock'] < 10.0 # Consumed!
|
|
print("PASS: test_chained_dependency_and_bottleneck")
|
|
|
|
def test_ipo_and_shares():
|
|
state = engine.get_initial_state()
|
|
state['lifetimeRevenue'] = 400000.0 # $400k
|
|
earned = engine.calculate_shares_earned(state['lifetimeRevenue'])
|
|
assert earned == 20 # 10 * sqrt(400k / 100k) = 10 * 2 = 20
|
|
|
|
# Apply IPO action
|
|
state = engine.apply_actions(state, [{'type': 'TRIGGER_IPO'}])
|
|
assert state['ventureShares'] == 20
|
|
assert state['ipoCount'] == 1
|
|
assert state['products']['wheat']['lines'] == 1
|
|
assert state['cash'] == 30.0
|
|
print("PASS: test_ipo_and_shares")
|
|
|
|
if __name__ == '__main__':
|
|
test_initial_state()
|
|
test_raw_production_and_logistics()
|
|
test_chained_dependency_and_bottleneck()
|
|
test_ipo_and_shares()
|
|
print("\nALL BACKEND ENGINE TESTS PASSED SUCCESSFULLY!")
|