Adding coverage on attention player

Author: moderouin

axelrod/tests/strategies/test_attention.py

@@ -42,6 +42,27 @@ def test_compute_features(self):
         self.assertEqual(features[3].item(), GameState.DefectCooperate)
         self.assertEqual(features[4].item(), GameState.CooperateDefect)
 
+    def test_compute_features_right_pad(self):
+        """Test that features are computed correctly."""
+        player = axl.MockPlayer(actions=[C, D, C, D])
+        opponent = axl.MockPlayer(actions=[D, C, C, D])
+        # Play the actions to populate history
+        match = axl.Match((player, opponent), turns=4)
+        match.play()
+
+        features = compute_features(player, opponent, True)
+
+        # Check the shape and type
+        self.assertIsInstance(features, torch.Tensor)
+        self.assertEqual(features.shape, (MEMORY_LENGTH + 1,))
+
+        # Check specific values (CLS token and game states)
+        self.assertEqual(features[0].item(), 0)  # CLS token
+        self.assertEqual(features[1].item(), GameState.DefectDefect)
+        self.assertEqual(features[2].item(), GameState.CooperateCooperate)
+        self.assertEqual(features[3].item(), GameState.DefectCooperate)
+        self.assertEqual(features[4].item(), GameState.CooperateDefect)
+
     def test_actions_to_game_state(self):
         """Test the mapping from actions to game states."""
         self.assertEqual(
@@ -52,6 +73,13 @@ def test_actions_to_game_state(self):
         self.assertEqual(actions_to_game_state(D, D), GameState.DefectDefect)
 
 
+class TestAttention(unittest.TestCase):
+    def test_initilization(self):
+        """Test that the model is initialized correctly."""
+        player = axl.Attention()
+        self.assertIsInstance(player.model, PlayerModel)
+
+
 class TestEvolvedAttention(TestPlayer):
     name = "EvolvedAttention"
     player = axl.EvolvedAttention
@@ -66,7 +94,7 @@ class TestEvolvedAttention(TestPlayer):
     }
 
     def test_model_initialization(self):
-        """Test that the model is initialized with pretrained weights."""
+        """Test that the model is initialized correctly."""
         player = self.player()
         self.assertIsInstance(player.model, PlayerModel)