зеркало из https://github.com/mozilla/bugbug.git
72 строки
2.2 KiB
Python
72 строки
2.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
||
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
||
|
|
# You can obtain one at http://mozilla.org/MPL/2.0/.
|
||
|
|
|
||
|
|
from bugbug.models.rcatype import RCATypeModel
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_rca_from_whiteboard():
|
||
|
|
model = RCATypeModel()
|
||
|
|
# Case 1: No rca
|
||
|
|
assert model.get_rca_from_whiteboard("[Whiteboard1][Not RCA type]") == []
|
||
|
|
# Case 2: RCA : A and RCA - A
|
||
|
|
assert model.get_rca_from_whiteboard("[RCA: cornercase]") == ["cornercase"]
|
||
|
|
assert model.get_rca_from_whiteboard("[rca - codingerror]") == ["codingerror"]
|
||
|
|
# Case 3: Multiple rca types
|
||
|
|
assert model.get_rca_from_whiteboard("[rca - cornercase][rca - codingerror]") == [
|
||
|
|
"cornercase",
|
||
|
|
"codingerror",
|
||
|
|
]
|
||
|
|
assert model.get_rca_from_whiteboard("[rca : systemerror][rca - codingerror]") == [
|
||
|
|
"systemerror",
|
||
|
|
"codingerror",
|
||
|
|
]
|
||
|
|
assert model.get_rca_from_whiteboard("[rca - cornercase][rca : testingerror]") == [
|
||
|
|
"cornercase",
|
||
|
|
"testingerror",
|
||
|
|
]
|
||
|
|
assert model.get_rca_from_whiteboard("[rca : cornercase][rca : codingerror]") == [
|
||
|
|
"cornercase",
|
||
|
|
"codingerror",
|
||
|
|
]
|
||
|
|
assert model.get_rca_from_whiteboard("[RCA: codingerror - syntaxerror]") == [
|
||
|
|
"codingerror"
|
||
|
|
]
|
||
|
|
# Case 4: subcategories enabled, with rca already present in the list
|
||
|
|
model = RCATypeModel(rca_subcategories_enabled=True)
|
||
|
|
assert model.get_rca_from_whiteboard("[RCA: codingerror - syntaxerror]") == [
|
||
|
|
"codingerror-syntaxerror"
|
||
|
|
]
|
||
|
|
assert model.get_rca_from_whiteboard(
|
||
|
|
"[RCA: codingerror - syntaxerror][rca: codingerror:logicalerror]"
|
||
|
|
) == ["codingerror-syntaxerror", "codingerror-logicalerror"]
|
||
|
|
# Case 5: subcategories enabled, with rca not present in list
|
||
|
|
assert model.get_rca_from_whiteboard("[RCA: codingerror - semanticerror]") == [
|
||
|
|
"codingerror-semanticerror"
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_labels():
|
||
|
|
model = RCATypeModel()
|
||
|
|
classes, _ = model.get_labels()
|
||
|
|
|
||
|
|
assert classes[1556846].tolist() == [
|
||
|
|
1.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
1.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
0.0,
|
||
|
|
]
|