feat: Enhance dashboard metrics and summary statistics
- Added new summary fields: variance, 5th percentile, 95th percentile, VaR (95%), and expected shortfall (95%) to the dashboard. - Updated the display logic for summary metrics to handle non-finite values gracefully. - Modified the chart rendering to include additional percentile points and tail risk metrics in tooltips. test: Introduce unit tests for consumption, costs, and other modules - Created a comprehensive test suite for consumption, costs, equipment, maintenance, production, reporting, and simulation modules. - Implemented fixtures for database setup and teardown using an in-memory SQLite database for isolated testing. - Added tests for creating, listing, and validating various entities, ensuring proper error handling and response validation. refactor: Consolidate parameter tests and remove deprecated files - Merged parameter-related tests into a new test file for better organization and clarity. - Removed the old parameter test file that was no longer in use. - Improved test coverage for parameter creation, listing, and validation scenarios. fix: Ensure proper validation and error handling in API endpoints - Added validation to reject negative amounts in consumption and production records. - Implemented checks to prevent duplicate scenario creation and ensure proper error messages are returned. - Enhanced reporting endpoint tests to validate input formats and expected outputs.
This commit is contained in:
@@ -1,42 +1,77 @@
|
||||
from uuid import uuid4
|
||||
|
||||
import pytest
|
||||
from fastapi.testclient import TestClient
|
||||
from main import app
|
||||
from config.database import Base, engine
|
||||
|
||||
# Setup and teardown
|
||||
|
||||
|
||||
def setup_module(module):
|
||||
Base.metadata.create_all(bind=engine)
|
||||
@pytest.fixture
|
||||
def client(api_client: TestClient) -> TestClient:
|
||||
return api_client
|
||||
|
||||
|
||||
def teardown_module(module):
|
||||
Base.metadata.drop_all(bind=engine)
|
||||
def _create_scenario(client: TestClient) -> int:
|
||||
payload = {
|
||||
"name": f"Equipment Scenario {uuid4()}",
|
||||
"description": "Scenario for equipment tests",
|
||||
}
|
||||
response = client.post("/api/scenarios/", json=payload)
|
||||
assert response.status_code == 200
|
||||
return response.json()["id"]
|
||||
|
||||
|
||||
client = TestClient(app)
|
||||
def test_create_equipment(client: TestClient) -> None:
|
||||
scenario_id = _create_scenario(client)
|
||||
payload = {
|
||||
"scenario_id": scenario_id,
|
||||
"name": "Excavator",
|
||||
"description": "Heavy machinery",
|
||||
}
|
||||
|
||||
response = client.post("/api/equipment/", json=payload)
|
||||
assert response.status_code == 200
|
||||
created = response.json()
|
||||
assert created["id"] > 0
|
||||
assert created["scenario_id"] == scenario_id
|
||||
assert created["name"] == "Excavator"
|
||||
assert created["description"] == "Heavy machinery"
|
||||
|
||||
|
||||
def test_create_and_list_equipment():
|
||||
# Create a scenario to attach equipment
|
||||
resp = client.post(
|
||||
"/api/scenarios/", json={"name": "EquipScenario", "description": "equipment scenario"}
|
||||
def test_list_equipment_filters_by_scenario(client: TestClient) -> None:
|
||||
target_scenario = _create_scenario(client)
|
||||
other_scenario = _create_scenario(client)
|
||||
|
||||
for scenario_id, name in [
|
||||
(target_scenario, "Bulldozer"),
|
||||
(target_scenario, "Loader"),
|
||||
(other_scenario, "Conveyor"),
|
||||
]:
|
||||
response = client.post(
|
||||
"/api/equipment/",
|
||||
json={
|
||||
"scenario_id": scenario_id,
|
||||
"name": name,
|
||||
"description": f"Equipment {name}",
|
||||
},
|
||||
)
|
||||
assert response.status_code == 200
|
||||
|
||||
list_response = client.get("/api/equipment/")
|
||||
assert list_response.status_code == 200
|
||||
items = [
|
||||
item
|
||||
for item in list_response.json()
|
||||
if item["scenario_id"] == target_scenario
|
||||
]
|
||||
assert {item["name"] for item in items} == {"Bulldozer", "Loader"}
|
||||
|
||||
|
||||
def test_create_equipment_requires_name(client: TestClient) -> None:
|
||||
scenario_id = _create_scenario(client)
|
||||
response = client.post(
|
||||
"/api/equipment/",
|
||||
json={
|
||||
"scenario_id": scenario_id,
|
||||
"description": "Missing name",
|
||||
},
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
scenario = resp.json()
|
||||
sid = scenario["id"]
|
||||
|
||||
# Create Equipment item
|
||||
eq_payload = {"scenario_id": sid, "name": "Excavator",
|
||||
"description": "Heavy machinery"}
|
||||
resp2 = client.post("/api/equipment/", json=eq_payload)
|
||||
assert resp2.status_code == 200
|
||||
eq = resp2.json()
|
||||
assert eq["scenario_id"] == sid
|
||||
assert eq["name"] == "Excavator"
|
||||
|
||||
# List Equipment items
|
||||
resp3 = client.get("/api/equipment/")
|
||||
assert resp3.status_code == 200
|
||||
data = resp3.json()
|
||||
assert any(item["name"] == "Excavator" and item["scenario_id"]
|
||||
== sid for item in data)
|
||||
assert response.status_code == 422
|
||||
|
||||
Reference in New Issue
Block a user