|
|
| import os
|
| import sys
|
| import unittest
|
|
|
| parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
|
|
|
|
|
| if parent_dir not in sys.path:
|
| sys.path.insert(0, parent_dir)
|
| from communicator import Synapse, SpikeEncoder
|
|
|
| class TestCommunicatorComponents(unittest.TestCase):
|
| def test_synapse_initialization(self):
|
| models = ["UI/UX", "Backend"]
|
| synapse = Synapse(models)
|
| self.assertEqual(synapse.weights, {"UI/UX": 0.5, "Backend": 0.5})
|
|
|
| def test_synapse_update_weights(self):
|
| models = ["UI/UX", "Backend"]
|
| synapse = Synapse(models, learning_rate=0.1, decay=0.9)
|
| synapse.update_weights(["UI/UX"])
|
| self.assertTrue(synapse.weights["UI/UX"] > 0.5)
|
| self.assertTrue(synapse.weights["Backend"] < 0.5)
|
|
|
| def test_spike_encoder_encode(self):
|
| encoder = SpikeEncoder(threshold=0.5)
|
| activations = {"UI/UX": 0.6, "Backend": 0.4}
|
| spikes = encoder.encode(activations)
|
| self.assertEqual(spikes, {"UI/UX": 1, "Backend": 0})
|
|
|
| def test_spike_encoder_decode(self):
|
| encoder = SpikeEncoder()
|
| spikes = {"UI/UX": 1, "Backend": 0}
|
| decoded = encoder.decode(spikes)
|
| self.assertEqual(decoded, {"UI/UX": 1.0, "Backend": 0.0})
|
|
|
| if __name__ == '__main__':
|
| unittest.main() |
