mirror of
https://github.com/WallabyLester/RBF-aPID-Controller.git
synced 2026-05-09 20:53:57 +08:00
Setting up aPID controller tests.
This commit is contained in:
@@ -4,5 +4,17 @@ import tensorflow as tf
|
||||
from aPID_tf import AdaptivePIDTf
|
||||
from RBF_tf import RBFAdaptiveModel
|
||||
|
||||
class TestAdaptivePIDTf(unittest.TestCase):
|
||||
def setUp(self):
|
||||
""" Set up RBFAdaptiveModel and AdaptivePIDTf class instances."""
|
||||
self.Kp = 7.0
|
||||
self.Ki = 0.5
|
||||
self.Kd = 0.01
|
||||
self.n_centers = 5
|
||||
self.input_dim = 3
|
||||
self.rbf_model = RBFAdaptiveModel(self.n_centers, self.input_dim)
|
||||
self.pid_controller = AdaptivePIDTf(self.Kp, self.Ki, self.Kd, self.rbf_model)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user