import unittest

from . import calculate_swi


class CalculateSwi(unittest.TestCase):
	def test_out_of_range(self):
		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):
		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)