def test_TAU_equation(self):
self.assertEqual(
eq_name(
get_necessary_2d_equations(['U', 'V', 'J', 'W', 'H', 'B', 'I'],
['TAU'], CHEZY_EQUATION)),
['M', 'US', 'TAU'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['U', 'V', 'M', 'Q', 'W', 'H'],
['TAU', 'Q', 'U', 'V'],
NIKURADSE_EQUATION)), ['US', 'TAU'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(
['U', 'V', 'M', 'US', 'Q', 'W', 'H'],
['I', 'TAU', 'Q', 'U', 'V'], STRICKLER_EQUATION)),
['I', 'TAU'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['U', 'V', 'M', 'US', 'W', 'H'],
['Q', 'TAU', 'US'],
MANNING_EQUATION)),
['I', 'J', 'Q', 'TAU'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(
['U', 'V', 'M', 'US', 'W', 'S', 'B'], ['Q', 'TAU'],
STRICKLER_EQUATION)), ['H', 'I', 'J', 'Q', 'TAU'])
def test_C_equation(self):
self.assertEqual(
eq_name(
get_necessary_2d_equations(['U', 'V', 'H', 'B'],
['U', 'C', 'H'], None)), ['C'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['U', 'S', 'Q', 'B'],
['U', 'C', 'H'], None)), ['H', 'C'])
def test_S_equation(self):
self.assertEqual(
eq_name(
get_necessary_2d_equations(['U', 'V', 'H', 'B'], ['U', 'S'],
None)), ['S'])
self.assertEqual(
eq_name(get_necessary_2d_equations(['H', 'Q', 'B'], ['S'], None)),
['S'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['Q', 'I', 'H', 'U', 'V', 'B'],
['S', 'U', 'H'], None)), ['S'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(
['U', 'B', 'J', 'I', 'Q', 'TAU', 'H', 'B'], ['H', 'S'],
None)), ['S'])
def test_no_equation(self):
self.assertEqual(
eq_name(get_necessary_2d_equations(['U', 'V'], ['U'], None)), [])
self.assertEqual(
eq_name(get_necessary_2d_equations(['U', 'Q'], ['Q'], None)), [])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['U', 'V', 'M', 'H', 'US'], ['US'],
None)), [])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['U', 'V', 'M', 'H', 'US'],
['US', 'U'], None)), [])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['U', 'V', 'M', 'H', 'US'],
['V', 'US', 'M'], None)), [])
self.assertEqual(
eq_name(
get_necessary_2d_equations(
['U', 'V', 'M', 'RANDOM NAME', 'H', 'US'], ['RANDOM NAME'],
None)), [])
self.assertEqual(
eq_name(
get_necessary_2d_equations(
['U', 'V', 'M', 'H', 'C', 'Q', 'F', 'US'], ['C'], None)),
[])
def test_B_equation(self):
self.assertEqual(
eq_name(
get_necessary_2d_equations(['HD', 'RB', 'H'], ['B', 'H'],
None)), ['B'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['HD', 'H', 'S', 'TAU'], ['B', 'S'],
None)), ['B'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['S', 'U', 'H', 'V', 'H'],
['S', 'B', 'H'], None)), ['B'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(
['DMAX', 'US', 'RB', 'H', 'Q', 'TAU', 'S', 'I'],
['H', 'B'], None)), ['B'])
def test_QS_equation(self):
self.assertEqual(
eq_name(
get_necessary_2d_equations(['HD', 'EF', 'B', 'DF'],
['B', 'QS'], None)), ['QS'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['EF', 'H', 'S', 'DF'], ['QS', 'S'],
None)), ['QS'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(
['QSX', 'EF', 'H', 'DF', 'QSY', 'B'], ['S', 'QS', 'H'],
None)), ['S', 'QS'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(
['DMAX', 'US', 'QSX', 'EF', 'Q', 'DF', 'S', 'B'],
['H', 'QS'], None)), ['H', 'QS'])
def test_FROPT_equation(self):
self.assertEqual(
eq_name(get_necessary_2d_equations(['US', 'MU'], ['FROTP'], None)),
['TAU', 'FROTP'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['U', 'MU', 'H', 'W', 'TAU', 'US'],
['FROTP'], None)), ['FROTP'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(
['U', 'V', 'J', 'W', 'H', 'MU', 'I'], ['FROTP'],
CHEZY_EQUATION)), ['M', 'US', 'TAU', 'FROTP'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(
['U', 'MU', 'M', 'B', 'Q', 'W', 'S'],
['FROTP', 'DMAX', 'H', 'Q', 'TAU', 'U', 'V'],
STRICKLER_EQUATION)), ['H', 'US', 'TAU', 'DMAX', 'FROTP'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['US', 'I', 'H', 'MU', 'V'],
['M', 'DMAX', 'Q', 'FROTP'], None)),
['M', 'J', 'Q', 'TAU', 'DMAX', 'FROTP'])
self.assertEqual(
eq_name(
get_necessary_2d_equations(['US', 'MU', 'H', 'U', 'V'],
['FROTP', 'M', 'DMAX', 'Q'], None)),
['M', 'I', 'J', 'Q', 'TAU', 'DMAX', 'FROTP'])