class_docstr: Tests for Scenic Optax utils.
class_docstr_tok: ["Tests","for","Scenic","Optax","utils","."]
docstr: Tests `_split_frozen` without a frozen schedule.
docstr_tok: ["Tests","`","_split_frozen","`","without","a","frozen","schedule","."]
code: def test__split_frozen_without_frozen(self):
      """Tests `_split_frozen` without a frozen schedule."""
      masks = [
          {"a": {"b": True, "d": False}, "c": False},
          {"a": {"b": False, "d": True}, "c": False},
      ]
      scheds = [1, 2]
      # pylint: disable=protected-access
      frozen_mask, masks, scheds = optax._split_frozen(masks, scheds)
      # pylint: enable=protected-access
  
      target_frozen_mask = {"a": {"b": False, "d": False}, "c": True}
      target_masks = [
          {"a": {"b": True, "d": False}, "c": False},
          {"a": {"b": False, "d": True}, "c": False},
      ]
      target_scheds = [1, 2]
      self.assertAllEqual(frozen_mask, target_frozen_mask)
      self.assertAllEqual(masks, target_masks)
      self.assertAllEqual(scheds, target_scheds)
  
imports: from typing import Optional
  from absl.testing import absltest
  from absl.testing import parameterized
  import ml_collections
  from scenic.train_lib import optax
  import tensorflow
label_desc: Use image node_id 5 for calling the OptaxTest obj's underlying member method code with example usage: obj.test__split_frozen_without_frozen() without return types