diff options
| -rw-r--r-- | szilagyi/tests.py | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/szilagyi/tests.py b/szilagyi/tests.py index 87b4dd4..bf59873 100644 --- a/szilagyi/tests.py +++ b/szilagyi/tests.py @@ -5,12 +5,27 @@ from . import calculate_swi class CalculateSwi(unittest.TestCase): def test_out_of_range(self): - for dt, depth in [(20, -10), (-10, 25000), (20, 6e+4), (41, 25000), (41, -10)]: + samples = [ + (20, -10), + (-10, 25000), + (20, 60000), + (41, 25000), + (41, -10), + ] + for dt, depth in samples: with self.assertRaises(ValueError): calculate_swi(dt, depth) def test_between(self): - for dt, depth, low, high in [(6.53, 3e+4, 1, 2)]: + samples = [ + (6.53, 30000, 1, 2), + (10.78, 20500, -1, 0), + (24.45, 2600, 0, 1), + (7.22, 1300, -10, -9), + (16.1, 6200, 0, 1), + (3, 20000, -4, -3), + ] + for dt, depth, low, high in samples: swi = calculate_swi(dt, depth) self.assertGreater(swi, low) self.assertLess(swi, high) |