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))