From 64a7812c170f5d46ef16a1517afddc7cd92c5240 Mon Sep 17 00:00:00 2001
From: Irit Katriel <1055913+iritkatriel@users.noreply.github.com>
Date: Wed, 11 Aug 2021 17:34:01 +0100
Subject: Update test__opcode and _Py_GetSpecializationStats with recent
 specialization stat changes (GH-27728)

---
 Lib/test/test__opcode.py | 6 ++++--
 Python/specialize.c      | 1 +
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/Lib/test/test__opcode.py b/Lib/test/test__opcode.py
index f6d8c73..6bbab53 100644
--- a/Lib/test/test__opcode.py
+++ b/Lib/test/test__opcode.py
@@ -82,10 +82,12 @@ class SpecializationStatsTests(unittest.TestCase):
             self.assertCountEqual(stats.keys(), specialized_opcodes)
             self.assertCountEqual(
                 stats['load_attr'].keys(),
-                stat_names + ['fails'])
+                stat_names + ['specialization_failure_kinds'])
             for sn in stat_names:
                 self.assertIsInstance(stats['load_attr'][sn], int)
-            self.assertIsInstance(stats['load_attr']['specialization_failure_kinds'], tuple)
+            self.assertIsInstance(
+                stats['load_attr']['specialization_failure_kinds'],
+                tuple)
             for v in stats['load_attr']['specialization_failure_kinds']:
                 self.assertIsInstance(v, int)
 
diff --git a/Python/specialize.c b/Python/specialize.c
index d0e45d7..f0d68f0 100644
--- a/Python/specialize.c
+++ b/Python/specialize.c
@@ -122,6 +122,7 @@ _Py_GetSpecializationStats(void) {
     err += add_stat_dict(stats, LOAD_ATTR, "load_attr");
     err += add_stat_dict(stats, LOAD_GLOBAL, "load_global");
     err += add_stat_dict(stats, BINARY_SUBSCR, "binary_subscr");
+    err += add_stat_dict(stats, STORE_ATTR, "store_attr");
     if (err < 0) {
         Py_DECREF(stats);
         return NULL;
-- 
cgit v0.12