update tests after new feature added

Author: GiacomoPope

tests/test_pkcs.py

@@ -161,7 +161,7 @@ def test_import_from_expanded(self):
         self.assertIs(ml_dsa, self.ml_dsa)
         self.assertEqual(sk, self.sk)
         self.assertEqual(seed, None)
-        self.assertEqual(pk, None)
+        self.assertEqual(pk, self.pk)
 
 
 @unittest.skipUnless(ECDSA_PRESENT, "requires ecdsa package")
@@ -1006,7 +1006,7 @@ def test_sk_sanity_expanded_only(self):
         self.assertEqual(self.ml_dsa, ml_dsa)
         self.assertEqual(self.sk, expanded)
         self.assertEqual(None, seed)
-        self.assertEqual(None, pk)
+        self.assertEqual(self.pk, pk)
 
     def test_sk_trailing_junk(self):
         enc = (