add test for is_partitioned_rewards_feature_enabled (#32158)

* add set_partitioned_rewards_enabled_for_test fn

* reveiws

---------

Co-authored-by: HaoranYi <[email protected]>

runtime/src/bank.rs

@@ -1492,7 +1492,8 @@ impl Bank {
 
     #[allow(dead_code)]
     fn is_partitioned_rewards_feature_enabled(&self) -> bool {
-        false // Will be feature later. It is convenient to have a constant fn at the moment.
+        self.feature_set
+            .is_active(&feature_set::enable_partitioned_epoch_reward::id())
     }
 
     #[allow(dead_code)]

runtime/src/bank/tests.rs

@@ -12604,6 +12604,17 @@ fn test_rewards_point_calculation_empty() {
     assert!(point_value.is_none());
 }
 
+/// Test partitioned_reward feature enable/disable
+#[test]
+fn test_is_partitioned_reward_enable() {
+    let (genesis_config, _mint_keypair) = create_genesis_config(1_000_000 * LAMPORTS_PER_SOL);
+
+    let mut bank = Bank::new_for_tests(&genesis_config);
+    assert!(!bank.is_partitioned_rewards_feature_enabled());
+    bank.activate_feature(&feature_set::enable_partitioned_epoch_reward::id());
+    assert!(bank.is_partitioned_rewards_feature_enabled());
+}
+
 #[test]
 fn test_deactivate_epoch_reward_status() {
     let (genesis_config, _mint_keypair) = create_genesis_config(1_000_000 * LAMPORTS_PER_SOL);