def test_prepare_map_arrays(self):
test_map = MS.Map()
test_map.prepare_map_arrays(1)
self.assertEqual(test_map.Light.shape[0], 7)
self.assertEqual(test_map.Light.shape[1], 7)
self.assertEqual(test_map.Dark.shape[0], 7)
self.assertEqual(test_map.Dark.shape[1], 7)
self.assertEqual(test_map.DifLight.shape[0], 7)
self.assertEqual(test_map.DifLight.shape[1], 7)
self.assertEqual(test_map.DifDark.shape[0], 7)
self.assertEqual(test_map.DifDark.shape[1], 7)
self.assertEqual(test_map.LDPressure.shape[0], 7)
self.assertEqual(test_map.LDPressure.shape[1], 7)
def test_inner_buffer_stencil(self):
# First, set up the map and the initial conditions for the tests.
test_map = MS.Map()
test_map.Light = 100 * np.ones([1, 9, 9])
test_map.Light = np.append(test_map.Light, np.zeros([1, 9, 9]), axis=0)
test_map.DifLight = 0.05 * np.ones([9, 9])
test_map.LDPressure = 200 * np.ones([1, 9, 9])
test_map.LDPressure = np.append(test_map.LDPressure,
np.zeros([1, 9, 9]),
axis=0)
time_step = 1
test_map.Light[time_step - 1, 4, 4] = 150
# Next, calculate the values for the stencil.
test_map.inner_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 4, 4)
test_map.inner_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 2, 4)
test_map.inner_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 3, 4)
test_map.inner_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 5, 4)
test_map.inner_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 6, 4)
test_map.inner_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 4, 2)
test_map.inner_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 4, 3)
test_map.inner_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 4, 5)
test_map.inner_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 4, 6)
test_map.inner_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 3, 3)
# Round the values to 3dp so that they can be easily asserted.
test_map.Light = test_map.Light.round(3)
# Lastly test the outputs of the stencil.
self.assertEqual(test_map.Light[1, 4, 4], 125)
self.assertEqual(test_map.Light[1, 2, 4], 99.583)
self.assertEqual(test_map.Light[1, 3, 4], 106.667)
self.assertEqual(test_map.Light[1, 5, 4], 106.667)
self.assertEqual(test_map.Light[1, 6, 4], 99.583)
self.assertEqual(test_map.Light[1, 4, 2], 99.583)
self.assertEqual(test_map.Light[1, 4, 3], 106.667)
self.assertEqual(test_map.Light[1, 4, 5], 106.667)
self.assertEqual(test_map.Light[1, 4, 6], 99.583)
self.assertEqual(test_map.Light[1, 3, 3], 100)
def test_initialise_values(self):
test_map = MS.Map()
test_map.prepare_map_arrays(1)
test_map.initialise_values()
self.assertEqual(test_map.Light[3, 3], 100)
self.assertEqual(test_map.Dark[3, 3], 100)
self.assertEqual(test_map.DifLight[3, 3], 1)
self.assertEqual(test_map.DifDark[3, 3], 1)
self.assertEqual(test_map.LDPressure[3, 3], 200)
test_map.initialise_values(50, 3)
self.assertEqual(test_map.Light[3, 3], 50)
self.assertEqual(test_map.Dark[3, 3], 50)
self.assertEqual(test_map.DifLight[3, 3], 3)
self.assertEqual(test_map.DifDark[3, 3], 3)
self.assertEqual(test_map.LDPressure[3, 3], 100)
def test_create_next_time_step(self):
test_map = MS.Map()
test_map.prepare_map_arrays(1)
test_map.create_next_time_step()
test_sizes1 = np.array([2, 7, 7])
for i in range(len(test_sizes1)):
self.assertEqual(test_map.Light.shape[i], test_sizes1[i])
self.assertEqual(test_map.Dark.shape[i], test_sizes1[i])
test_map.create_next_time_step()
test_sizes2 = np.array([3, 7, 7])
for i in range(len(test_sizes2)):
self.assertEqual(test_map.Light.shape[i], test_sizes2[i])
self.assertEqual(test_map.Dark.shape[i], test_sizes2[i])
test_map.create_next_time_step()
test_map.create_next_time_step()
test_map.create_next_time_step()
test_sizes3 = np.array([6, 7, 7])
for i in range(len(test_sizes2)):
self.assertEqual(test_map.Light.shape[i], test_sizes3[i])
self.assertEqual(test_map.Dark.shape[i], test_sizes3[i])
# Author: Jack Adams
# Date Started: 18/05/27
# Last Updated: 18/06/1
# This script is used to run the finite element schemes over time.
import MapStructures
import numpy as np
aramour = MapStructures.Map()
width = 15
aramour.prepare_map_arrays(width)
aramour.initialise_values(100, 0.04)
aramour.create_next_time_step()
aramour.Light[0, 11, 11] = 150
aramour.LDPressure[0, 11, 11] = 250
for tstep in range(1, 100):
aramour.calculate_next_time_step(aramour.Light, aramour.DifLight,
aramour.LDPressure, tstep, width)
aramour.calculate_next_time_step(aramour.Dark, aramour.DifDark,
aramour.LDPressure, tstep, width)
aramour.generate_BCs(aramour.Light, tstep, width)
aramour.generate_BCs(aramour.Dark, tstep, width)
aramour.LD_forcing_functions(aramour.Light, aramour.Dark, tstep, width)
aramour.update_pressure(aramour.Light, aramour.Dark, aramour.LDPressure,
tstep, width)
aramour.create_next_time_step()
print("Done!")
def test_roi_stencil(self):
# First, set up the map and the initial conditions for the tests.
test_map = MS.Map()
test_map.Light = 100 * np.ones([1, 13, 13])
test_map.Light = np.append(test_map.Light,
np.zeros([1, 13, 13]),
axis=0)
test_map.DifLight = 0.05 * np.ones([13, 13])
test_map.LDPressure = 200 * np.ones([1, 13, 13])
test_map.LDPressure = np.append(test_map.LDPressure,
np.zeros([1, 13, 13]),
axis=0)
time_step = 1
test_map.Light[time_step - 1, 6, 6] = 150
# Next, calculate the values for the stencil.
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 6, 6)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 3, 6)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 4, 6)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 5, 6)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 7, 6)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 8, 6)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 9, 6)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 6, 3)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 6, 4)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 6, 5)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 6, 7)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 6, 8)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 6, 9)
test_map.roi_stencil(test_map.Light, test_map.DifLight,
test_map.LDPressure, time_step, 5, 5)
# Round the values to 3dp so that they can be easily asserted.
test_map.Light = test_map.Light.round(3)
# Lastly test the outputs of the stencil.
self.assertEqual(test_map.Light[1, 6, 6], 122.778)
self.assertEqual(test_map.Light[1, 3, 6], 100.056)
self.assertEqual(test_map.Light[1, 4, 6], 99.250)
self.assertEqual(test_map.Light[1, 5, 6], 107.500)
self.assertEqual(test_map.Light[1, 7, 6], 107.500)
self.assertEqual(test_map.Light[1, 8, 6], 99.250)
self.assertEqual(test_map.Light[1, 9, 6], 100.056)
self.assertEqual(test_map.Light[1, 6, 3], 100.056)
self.assertEqual(test_map.Light[1, 6, 4], 99.250)
self.assertEqual(test_map.Light[1, 6, 5], 107.500)
self.assertEqual(test_map.Light[1, 6, 7], 107.500)
self.assertEqual(test_map.Light[1, 6, 8], 99.250)
self.assertEqual(test_map.Light[1, 6, 9], 100.056)
self.assertEqual(test_map.Light[1, 5, 5], 100.000)
def test_map(self):
test_map = MS.Map()
self.assertEqual(test_map.Light, None)
self.assertEqual(test_map.Dark, None)
self.assertEqual(test_map.DifLight, None)
self.assertEqual(test_map.DifDark, None)