code: def test_basic(self) -> None:
      self.assertEqual(
          list(
              batch_edges_mix_relation_types(
                  EdgeList(
                      EntityList.from_tensor(
                          torch.tensor(
                              [93, 24, 13, 31, 70, 66, 77, 38, 5, 5],
                              dtype=torch.long,
                          )
                      ),
                      EntityList.from_tensor(
                          torch.tensor(
                              [90, 75, 9, 25, 23, 31, 49, 64, 42, 50],
                              dtype=torch.long,
                          )
                      ),
                      torch.tensor(
                          [1, 0, 0, 1, 2, 2, 0, 0, 2, 2],
                          dtype=torch.long,
                      ),
                  ),
                  batch_size=4,
              )
          ),
          [
              EdgeList(
                  EntityList.from_tensor(
                      torch.tensor([93, 24, 13, 31], dtype=torch.long)
                  ),
                  EntityList.from_tensor(
                      torch.tensor([90, 75, 9, 25], dtype=torch.long)
                  ),
                  torch.tensor([1, 0, 0, 1], dtype=torch.long),
              ),
              EdgeList(
                  EntityList.from_tensor(
                      torch.tensor([70, 66, 77, 38], dtype=torch.long)
                  ),
                  EntityList.from_tensor(
                      torch.tensor([23, 31, 49, 64], dtype=torch.long)
                  ),
                  torch.tensor([2, 2, 0, 0], dtype=torch.long),
              ),
              EdgeList(
                  EntityList.from_tensor(
                      torch.tensor([5, 5], dtype=torch.long)
                  ),
                  EntityList.from_tensor(
                      torch.tensor([42, 50], dtype=torch.long)
                  ),
                  torch.tensor([2, 2], dtype=torch.long),
              ),
          ],
      )
  
file_path: PyTorch-BigGraph/test/test_batching.py
imports: from collections import defaultdict
  from unittest import main, TestCase
  import torch
  from torchbiggraph.batching import batch_edges_group_by_relation_type, batch_edges_mix_relation_types, group_by_relation_type
  from torchbiggraph.edgelist import EdgeList
  from torchbiggraph.entitylist import EntityList
label_desc: Use image node_id 1 for calling the TestBatchEdgesMixRelationTypes obj's underlying member method code with example usage: obj.test_basic() without return types