def test_next_layer_outputs_correct_vector(self):
digit = Digit()
in_vector = [[0.5, 0.75]]
wieghts = [
[2, 1], #bias weights
[3, 4],
[0, 1]
]
expected_output = [[0.970687702262056, 0.9770225734624428]]
actual = digit.calc_next_layer(in_vector, wieghts)
self.assertEqual(actual, expected_output)
def test_make_network_creates_correct_wieght_array_for_given_nodes_in_layer(
self):
digit = Digit()
digit.file_wieghts_location = "text_make_network.txt"
num_nodes_in_layer = [2, 3, 1]
wieghts0 = [[1.0, 1.0, 1.0], [1.0, 1.0, 1.0], [1.0, 1.0, 1.0]]
wieghts1 = [[1.0], [1.0], [1.0], [1.0]]
expected_my_wieghts = [wieghts0, wieghts1]
digit.make_network(num_nodes_in_layer, 1.0)
actual = digit.mywieghts3D
self.assertEqual(actual, expected_my_wieghts)
def test_save_wieght_and_read_wieght_work_together_2(self):
digit = Digit()
digit.file_wieghts_location = "test_wieghts2.txt"
wieghts0 = [[1.2, 2.0], [4.0, 2.0]]
wieghts1 = [[5.0, 6.3, 7.0, 8.0], [2.0, 3.0, 1.0, 4.0],
[2.9, 7.0, 9.0, 9.0]]
my_wieghts = [wieghts0, wieghts1]
digit.mywieghts3D = [wieghts0, wieghts1]
digit.save_wieghts()
digit.mywieghts3D = [[[]]] #to make sure it doesnt cheat test
digit.read_wieghts()
actual = digit.mywieghts3D
self.assertEqual(actual, my_wieghts)
def test_find_output_from_input_calculates_correct_output_for_one_input(
self):
digit = Digit()
nodes_in_layer = [3, 2, 2]
digit.make_network(nodes_in_layer)
input_data = [[1.0, 2.0, 3.0]]
digit.records = [input_data, []]
expected = [[0.9524916507608503, 0.9524916507608503]]
digit.find_output_from_input()
actual = digit.calc_outputs
self.assertEqual(actual, expected)
def test_read_data_loads_records_correctly(self):
digit = Digit()
digit.file_records_location = "needed_data/test_read_data.txt"
num_input = 4
num_output = 2
num_records = 3
expected = [
[
[5.0, 2.0, 4.0, 3.0], #Input
[2.0, 9.0, 1.0, 5.0],
[5.0, 2.0, 4.0, 3.0]
],
[
[0.5, 0.3], #Output
[0.7, 0.4],
[0.5, 0.3]
]
]
digit.read_data(num_input, num_output, num_records)
actual = digit.records
self.assertEqual(actual, expected)
def test_find_output_from_input_calculates_correct_output_for_mult_inputs(
self):
digit = Digit()
nodes_in_layer = [3, 2, 2]
digit.make_network(nodes_in_layer)
input_data = [[1.0, 2.0, 3.0], [2.0, 1.0, 3.0], [5.0, 4.0, 3.0],
[1.0, 0.5, 2.0]]
digit.records = [input_data, []]
expected = [[0.9524916507608503, 0.9524916507608503],
[0.9524916507608503, 0.9524916507608503],
[0.9525738314274983, 0.9525738314274983],
[0.9515713940433558, 0.9515713940433558]]
digit.find_output_from_input()
actual = digit.calc_outputs
self.assertEqual(actual, expected)