11import pytest
22import math
33import random
4- from utils import DataFile
4+ from utils import open_data
55from learning import *
66
77
@@ -18,27 +18,31 @@ def test_euclidean():
1818 distance = euclidean_distance ([0 , 0 , 0 ], [0 , 0 , 0 ])
1919 assert distance == 0
2020
21+
2122def test_rms_error ():
2223 assert rms_error ([2 , 2 ], [2 , 2 ]) == 0
2324 assert rms_error ((0 , 0 ), (0 , 1 )) == math .sqrt (0.5 )
2425 assert rms_error ((1 , 0 ), (0 , 1 )) == 1
2526 assert rms_error ((0 , 0 ), (0 , - 1 )) == math .sqrt (0.5 )
2627 assert rms_error ((0 , 0.5 ), (0 , - 0.5 )) == math .sqrt (0.5 )
2728
29+
2830def test_manhattan_distance ():
2931 assert manhattan_distance ([2 , 2 ], [2 , 2 ]) == 0
3032 assert manhattan_distance ([0 , 0 ], [0 , 1 ]) == 1
3133 assert manhattan_distance ([1 , 0 ], [0 , 1 ]) == 2
3234 assert manhattan_distance ([0 , 0 ], [0 , - 1 ]) == 1
3335 assert manhattan_distance ([0 , 0.5 ], [0 , - 0.5 ]) == 1
3436
37+
3538def test_mean_boolean_error ():
3639 assert mean_boolean_error ([1 , 1 ], [0 , 0 ]) == 1
3740 assert mean_boolean_error ([0 , 1 ], [1 , 0 ]) == 1
3841 assert mean_boolean_error ([1 , 1 ], [0 , 1 ]) == 0.5
3942 assert mean_boolean_error ([0 , 0 ], [0 , 0 ]) == 0
4043 assert mean_boolean_error ([1 , 1 ], [1 , 1 ]) == 0
4144
45+
4246def test_mean_error ():
4347 assert mean_error ([2 , 2 ], [2 , 2 ]) == 0
4448 assert mean_error ([0 , 0 ], [0 , 1 ]) == 0.5
@@ -53,7 +57,7 @@ def test_exclude():
5357
5458
5559def test_parse_csv ():
56- Iris = DataFile ('iris.csv' ).read ()
60+ Iris = open_data ('iris.csv' ).read ()
5761 assert parse_csv (Iris )[0 ] == [5.1 , 3.5 , 1.4 , 0.2 , 'setosa' ]
5862
5963
0 commit comments