def METHOD_NAME(self):
    labels = np.asarray([0, 0, 0, 0, 0, 1, 1, 1, 1, 1])
    dict_a = {'variable_one': np.random.randn(10),
              'variable_two': np.random.randn(10, 10)}
    dict_a_1 = PhotonDataHelper.index_dict(dict_a, labels == 0)
    dict_a_2 = PhotonDataHelper.index_dict(dict_a, labels == 1)
    self.assertEqual(len(dict_a_1['variable_one']), 5)
    self.assertEqual(dict_a_2['variable_two'].shape, (5, 10))