code: def test_round_indicator(self):
      indicator = restart_query.PeriodicTimeRestartIndicator(
          period_seconds=3600 * 23.5
      )
      # TODO(b/193679963): use `tf.timestamp` as the default of a member of
      # the `PeriodicTimeRestartIndicator` to unroll the mock test.
      return_time = tf.Variable(
          1627018868.452365
      )  # 22:41pm PST 5:41am UTC, July 22, initialize
      with unittest.mock.patch.object(
          tf, "timestamp", return_value=return_time
      ) as mock_func:
          time_stamps = [
              1627022468.452365,  # 23:41pm PST 5:41am UTC, July 22, 1 hr, False
              1627105268.452365,  # 22:41pm PST 5:41am UTC, July 23, 1 day, True
              1627112468.452365,  # 2 hr after restart, False
              1627189508.452365,  # 23.4 hr after restart, False
              1627189904.452365,  # 23.51 hr after restart, True
          ]
          expected_values = [False, True, False, False, True]
          state = indicator.initialize()
          for v, t in zip(expected_values, time_stamps):
              return_time.assign(t)
              mock_func.return_value = return_time
              flag, state = indicator.next(state)
              self.assertEqual(v, flag.numpy())
  
imports: import unittest
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_privacy.privacy.dp_query import restart_query
  from tensorflow_privacy.privacy.dp_query import tree_aggregation_query
label_desc: Use image node_id 2 for calling the TimeRestartIndicatorTest obj's underlying member method code with example usage: obj.test_round_indicator() without return types