code: def test_blockpartition_zeros():
      assert blockpartition.solve([0, 0], partitions=2) == [[0], [0]]
  
imports: import time
  import pytest
  import torch
  from torch import nn
  from fairscale.nn.pipe.balance import balance_by_size, balance_by_time, blockpartition
  from fairscale.nn.pipe.balance.profile import layerwise_sandbox
label_desc: Use image node_id 2 for calling a global function with example usage: test_blockpartition_zeros() without return types