82 lines
4 KiB
Python
82 lines
4 KiB
Python
import unittest
|
|
import pytest
|
|
from conditionals import (is_criticality_balanced,
|
|
reactor_efficiency,
|
|
fail_safe)
|
|
|
|
|
|
class MeltdownMitigationTest(unittest.TestCase):
|
|
"""Test cases for Meltdown mitigation exercise.
|
|
"""
|
|
|
|
@pytest.mark.task(taskno=1)
|
|
def test_is_criticality_balanced(self):
|
|
"""Testing border cases around typical points.
|
|
|
|
T, n == (800, 500), (625, 800), (500, 1000), etc.
|
|
|
|
"""
|
|
|
|
test_data = ((750, 650, True), (799, 501, True), (500, 600, True),
|
|
(1000, 800, False), (800, 500, False), (800, 500.01, False),
|
|
(799.99, 500, False), (500.01, 999.99, False), (625, 800, False),
|
|
(625.99, 800, False), (625.01, 799.99, False), (799.99, 500.01, True),
|
|
(624.99, 799.99, True), (500, 1000, False), (500.01, 1000, False),
|
|
(499.99, 1000, True))
|
|
|
|
for variant, data in enumerate(test_data, start=1):
|
|
temp, neutrons_emitted, expected = data
|
|
with self.subTest(f'variation #{variant}', temp=temp, neutrons_emitted=neutrons_emitted, expected=expected):
|
|
|
|
# pylint: disable=assignment-from-no-return
|
|
actual_result = is_criticality_balanced(temp, neutrons_emitted)
|
|
failure_message = (f'Called is_criticality_balanced({temp}, {neutrons_emitted}). '
|
|
f' The function returned {actual_result}, '
|
|
f'but the test expected {expected} as the return value.')
|
|
|
|
self.assertEqual(actual_result, expected, failure_message)
|
|
|
|
@pytest.mark.task(taskno=2)
|
|
def test_reactor_efficiency(self):
|
|
voltage = 10
|
|
theoretical_max_power = 10000
|
|
|
|
# The numbers are chosen so that current == 10 x percentage
|
|
test_data = ((1000, 'green'), (999, 'green'), (800, 'green'),
|
|
(799, 'orange'), (700, 'orange'), (600, 'orange'),
|
|
(599, 'red'), (560, 'red'), (400, 'red'), (300, 'red'),
|
|
(299, 'black'), (200, 'black'), (0, 'black'))
|
|
|
|
for variant, data in enumerate(test_data, start=1):
|
|
current, expected = data
|
|
with self.subTest(f'variation #{variant}', voltage=voltage, current=current,
|
|
theoretical_max_power=theoretical_max_power, expected=expected):
|
|
|
|
# pylint: disable=assignment-from-no-return
|
|
actual_result = reactor_efficiency(voltage, current, theoretical_max_power)
|
|
failure_message =(f'Called reactor_efficiency({voltage}, {current}, {theoretical_max_power}). '
|
|
f'The function returned {actual_result}, '
|
|
f'but the test expected {expected} as the return value.')
|
|
|
|
self.assertEqual(actual_result, expected, failure_message)
|
|
|
|
@pytest.mark.task(taskno=3)
|
|
def test_fail_safe(self):
|
|
temp = 10
|
|
threshold = 10000
|
|
test_data = ((399, 'LOW'), (300, 'LOW'), (1, 'LOW'),
|
|
(0, 'LOW'), (901, 'NORMAL'), (1000, 'NORMAL'),
|
|
(1099, 'NORMAL'), (899, 'LOW'), (700, 'LOW'),
|
|
(400, 'LOW'), (1101, 'DANGER'), (1200, 'DANGER'))
|
|
|
|
for variant, (neutrons_per_second, expected) in enumerate(test_data, start=1):
|
|
with self.subTest(f'variation #{variant}', temp=temp, neutrons_per_second=neutrons_per_second,
|
|
threshold=threshold, expected=expected):
|
|
|
|
# pylint: disable=assignment-from-no-return
|
|
actual_result = fail_safe(temp, neutrons_per_second, threshold)
|
|
failure_message = (f'Called fail_safe({temp}, {neutrons_per_second}, {threshold}). '
|
|
f'The function returned {actual_result}, '
|
|
f'but the test expected {expected} as the return value.')
|
|
|
|
self.assertEqual(actual_result, expected, failure_message)
|