def test_psi(): psi = [[0 for col in range(5)] for row in range(5)] expected = [[0 for col in range(5)] for row in range(5)] # Before: # 0 0 1 0 0 # 0 0 1 0 0 # 0 0 1 1 1 # 0 0 0 0 0 # 0 0 0 0 0 psi[0][2] = 1 psi[1][2] = 1 psi[2][2] = 1 psi[2][3] = 1 psi[2][4] = 1 # Expect: # 0 0 1 0 0 # 0 0.25 0.5 0.5 0 # 0 0.25 0.5 0.5 1 # 0 0 0.25 0.25 0 # 0 0 0 0 0 expected[0][2] = 1 expected[1][1] = 0.25 expected[1][2] = 0.5 expected[1][3] = 0.5 expected[2][1] = 0.25 expected[2][2] = 0.5 expected[2][3] = 0.5 expected[2][4] = 1 expected[3][2] = 0.25 expected[3][3] = 0.25 update_stream_function(3, 3, psi) np.testing.assert_allclose(expected, psi, rtol=0, atol=0.01)
def test_empty_psi(): psi = [] update_stream_function(0, 0, psi) assert_equal([], psi, 2)
def test_ones_psi(): psi = [[1 for col in range(5)] for row in range(5)] expected = [[1 for col in range(5)] for row in range(5)] update_stream_function(3, 3, psi) np.testing.assert_allclose(expected, psi, rtol=0, atol=0.01)
def test_zero_psi(): psi = [[0 for col in range(3)] for row in range(3)] expected = [[0 for col in range(3)] for row in range(3)] update_stream_function(1, 1, psi) np.testing.assert_allclose(expected, psi, rtol=0, atol=0.01)