Mirrors_Framework/RBF-aPID-Controller
1
0
Fork 0
mirror of https://github.com/WallabyLester/RBF-aPID-Controller.git synced 2026-05-09 20:53:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added test for predict function.

Browse Source
This commit is contained in:
Andru Liu
2024-10-09 15:15:46 -07:00
parent b2efa77059
commit 4e11dd8c76
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
NP_Implementation/tests/test_RBF_np.py
+14
View File
@@ -18,5 +18,19 @@ class TestRBFNetwork(unittest.TestCase):
output = self.rbf_network.gaussian(self.x, center)
self.assertAlmostEqual(output, expected_output, places=5)
def test_predict(self):
"""Test the predict function."""
output_before = self.rbf_network.predict(self.x)
self.assertIsInstance(output_before, float)
[self.assertNotAlmostEqual(self.rbf_network.weights[i], self.rbf_network.weights[i+1])
for i in range(len(self.rbf_network.weights)-1)]
target = 1.0
self.rbf_network.train(self.x, target)
output_after = self.rbf_network.predict(self.x)
self.assertIsInstance(output_after, float)
self.assertNotEqual(output_before, output_after)
if __name__ == "__main__":
unittest.main()