Mirrors_Framework/RBF-aPID-Controller
1
0
Fork 0
mirror of https://github.com/WallabyLester/RBF-aPID-Controller.git synced 2026-06-01 03:56:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added test for output shape of the model.

Browse Source
This commit is contained in:
Andru Liu
2024-10-09 19:48:23 -07:00
parent 32dae03ad1
commit aaeb3a4507
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TF_Implementation/test/test_RBF_tf.py
+7
View File
@@ -36,6 +36,13 @@ class TestRBFAdaptiveModel(unittest.TestCase):
self.assertIsInstance(self.model.rbf_layer, RBFLayer) self.assertIsInstance(self.model.rbf_layer, RBFLayer)
self.assertEqual(self.model.output_layer.units, 1) self.assertEqual(self.model.output_layer.units, 1)
def test_call_output_shape(self):
""" Test the call method for output shape."""
inputs = tf.random.normal((3, self.input_dim))
output = self.model(inputs)
self.assertEqual(output.shape, (3, 1))
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()