def test_mapper1_case_1(self):
# empty arc map
arc_map = k2host.IntArray1.create_array_with_size(0)
mapper = k2host.AuxLabels1Mapper(self.aux_labels_in, arc_map)
aux_size = k2host.IntArray2Size()
mapper.get_sizes(aux_size)
self.assertEqual(aux_size.size1, 0)
self.assertEqual(aux_size.size2, 0)
labels_out = k2host.AuxLabels.create_array_with_size(aux_size)
mapper.get_output(labels_out)
self.assertTrue(labels_out.empty())
def test_mapper1_case_2(self):
arc_map = k2host.IntArray1(torch.IntTensor([2, 0, 3]))
mapper = k2host.AuxLabels1Mapper(self.aux_labels_in, arc_map)
aux_size = k2host.IntArray2Size()
mapper.get_sizes(aux_size)
labels_out = k2host.AuxLabels.create_array_with_size(aux_size)
mapper.get_output(labels_out)
self.assertEqual(aux_size.size1, 3)
self.assertEqual(aux_size.size2, 5)
expected_indexes = torch.IntTensor([0, 3, 4, 5])
expected_data = torch.IntTensor([4, 5, 6, 1, 7])
self.assertTrue(torch.equal(labels_out.indexes, expected_indexes))
self.assertTrue(torch.equal(labels_out.data, expected_data))