def test_from_file_with_commas(self): s = """ 1,2,3, -1, -1, 1, 1, 2, 3 """ buffer = StringIO(s) anno = AnnotationsContainer._from_file_object(buffer) expected = np.array([[0, 1, 2], [MV, MV, 0], [0, 1, 2]]) np.testing.assert_equal(anno.annotations, expected)
def test_from_file_with_commas(self): s = """ 1,2,3, -1, -1, 1, 1, 2, 3 """ buffer = StringIO(s) anno = AnnotationsContainer._from_file_object(buffer) expected = np.array([ [0, 1, 2], [MV, MV, 0], [0, 1, 2] ]) np.testing.assert_equal(anno.annotations, expected)
def test_from_file_numerical_labels(self): s = """ 1 -1 2 2 1 3 -1 -1 3 """ buffer = StringIO(s) anno = AnnotationsContainer._from_file_object(buffer) expected = np.array([[0, MV, 1], [1, 0, 2], [MV, MV, 2]]) np.testing.assert_equal(anno.annotations, expected) self.assertEqual(anno.nclasses, 3) self.assertEqual(anno.labels, ['1', '2', '3']) self.assertEqual(anno.missing_values, ['-1'])
def test_from_file_string_labels(self): s = """ D B A A B C A * C A B B D D * """ buffer = StringIO(s) anno = AnnotationsContainer._from_file_object(buffer) self.assertEqual(anno.nclasses, 4) self.assertEqual(anno.nannotators, 5) # labels should be sorted self.assertEqual(anno.labels, ['A', 'B', 'C', 'D']) self.assertEqual(anno.missing_values, ['*']) expected = np.array( [[3, 1, 0, 0, 1], [2, 0, MV, 2, 0], [1, 1, 3, 3, MV]], dtype=int) np.testing.assert_equal(anno.annotations, expected)
def test_from_file_numerical_labels(self): s = """ 1 -1 2 2 1 3 -1 -1 3 """ buffer = StringIO(s) anno = AnnotationsContainer._from_file_object(buffer) expected = np.array([ [0, MV, 1], [1, 0, 2], [MV, MV, 2] ]) np.testing.assert_equal(anno.annotations, expected) self.assertEqual(anno.nclasses, 3) self.assertEqual(anno.labels, ['1', '2', '3']) self.assertEqual(anno.missing_values, ['-1'])
def test_from_file_string_labels(self): s = """ D B A A B C A * C A B B D D * """ buffer = StringIO(s) anno = AnnotationsContainer._from_file_object(buffer) self.assertEqual(anno.nclasses, 4) self.assertEqual(anno.nannotators, 5) # labels should be sorted self.assertEqual(anno.labels, ['A', 'B', 'C', 'D']) self.assertEqual(anno.missing_values, ['*']) expected = np.array([ [3, 1, 0, 0, 1], [2, 0, MV, 2, 0], [1, 1, 3, 3, MV] ], dtype=int) np.testing.assert_equal(anno.annotations, expected)