Mirrors_Framework/RBF-aPID-Controller
1
0
Fork 0
mirror of https://github.com/WallabyLester/RBF-aPID-Controller.git synced 2026-05-10 04:58:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add test for layer call.

Browse Source
This commit is contained in:
Andru Liu
2024-10-09 19:35:49 -07:00
parent 0e2908525d
commit effb96a9a2
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TF_Implementation/test/test_RBF_tf.py
+8
View File
@@ -16,6 +16,14 @@ class TestRBFLayer(unittest.TestCase):
self.assertEqual(self.rbf_layer.centers.shape, (self.n_centers, self.input_dim))
self.assertEqual(self.rbf_layer.sigmas.shape, (self.n_centers,))
def test_call(self):
""" Test the call method."""
inputs = tf.random.normal((3, self.input_dim))
output = self.rbf_layer(inputs)
self.assertEqual(output.shape, (3, self.n_centers))
self.assertTrue(tf.reduce_all(output >= 0))
if __name__ == '__main__':
unittest.main()