Commit 8e8faa12 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add bigger tolerance

parent 8ab17fe2

test/generic/test_multilabel_timm.py

+21 −21
Original line number Diff line number Diff line
@@ -33,9 +33,9 @@ class TestGenericMultilabelTIMM: 
            'medium_breasts': 0.4368951916694641, 'arms_up': 0.4188764989376068, 'hair_intakes': 0.394428551197052,

            'thighs': 0.33927807211875916, 'cleft_of_venus': 0.31293246150016785, 'collarbone': 0.30162444710731506,

            'arms_behind_head': 0.2740631699562073, 'clitoris': 0.1854049563407898

        })

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858})

        assert rating == pytest.approx({'explicit': 0.9496212005615234})

        }, abs=1e-2)

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858}, abs=1e-2)

        assert rating == pytest.approx({'explicit': 0.9496212005615234}, abs=1e-2)



    def test_multilabel_timm_predict_noname(self):

        general, character, rating = multilabel_timm_predict(
@@ -54,9 +54,9 @@ class TestGenericMultilabelTIMM: 
            'medium_breasts': 0.4368951916694641, 'arms_up': 0.4188764989376068, 'hair_intakes': 0.394428551197052,

            'thighs': 0.33927807211875916, 'cleft_of_venus': 0.31293246150016785, 'collarbone': 0.30162444710731506,

            'arms_behind_head': 0.2740631699562073, 'clitoris': 0.1854049563407898

        })

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858})

        assert rating == pytest.approx({'explicit': 0.9496212005615234})

        }, abs=1e-2)

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858}, abs=1e-2)

        assert rating == pytest.approx({'explicit': 0.9496212005615234}, abs=1e-2)



    def test_multilabel_timm_predict_with_category_thresholds(self):

        general, character, rating = multilabel_timm_predict(
@@ -76,9 +76,9 @@ class TestGenericMultilabelTIMM: 
            'anus': 0.5988802313804626, 'hair_between_eyes': 0.5209077596664429, 'closed_mouth': 0.5168408155441284,

            'medium_breasts': 0.4368951916694641, 'arms_up': 0.4188764989376068, 'hair_intakes': 0.394428551197052,

            'thighs': 0.33927807211875916, 'cleft_of_venus': 0.31293246150016785, 'collarbone': 0.30162444710731506

        })

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858})

        assert rating == pytest.approx({'explicit': 0.9496212005615234})

        }, abs=1e-2)

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858}, abs=1e-2)

        assert rating == pytest.approx({'explicit': 0.9496212005615234}, abs=1e-2)



    def test_multilabel_timm_predict_with_category_thresholds_cateid(self):

        general, character, rating = multilabel_timm_predict(
@@ -98,9 +98,9 @@ class TestGenericMultilabelTIMM: 
            'anus': 0.5988802313804626, 'hair_between_eyes': 0.5209077596664429, 'closed_mouth': 0.5168408155441284,

            'medium_breasts': 0.4368951916694641, 'arms_up': 0.4188764989376068, 'hair_intakes': 0.394428551197052,

            'thighs': 0.33927807211875916, 'cleft_of_venus': 0.31293246150016785, 'collarbone': 0.30162444710731506

        })

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858})

        assert rating == pytest.approx({'explicit': 0.9496212005615234})

        }, abs=1e-2)

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858}, abs=1e-2)

        assert rating == pytest.approx({'explicit': 0.9496212005615234}, abs=1e-2)



    def test_multilabel_timm_predict_with_category_thresholds_cateid_generic_thresholds(self):

        general, character, rating = multilabel_timm_predict(
@@ -120,9 +120,9 @@ class TestGenericMultilabelTIMM: 
            'anus': 0.5988802313804626, 'hair_between_eyes': 0.5209077596664429, 'closed_mouth': 0.5168408155441284,

            'medium_breasts': 0.4368951916694641, 'arms_up': 0.4188764989376068, 'hair_intakes': 0.394428551197052,

            'thighs': 0.33927807211875916, 'cleft_of_venus': 0.31293246150016785, 'collarbone': 0.30162444710731506

        })

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858})

        assert rating == pytest.approx({'explicit': 0.9496212005615234})

        }, abs=1e-2)

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858}, abs=1e-2)

        assert rating == pytest.approx({'explicit': 0.9496212005615234}, abs=1e-2)



    def test_multilabel_timm_predict_with_category_thresholds_cateid_no_tagth(self):

        general, character, rating = multilabel_timm_predict(
@@ -143,9 +143,9 @@ class TestGenericMultilabelTIMM: 
            'anus': 0.5988802313804626, 'hair_between_eyes': 0.5209077596664429, 'closed_mouth': 0.5168408155441284,

            'medium_breasts': 0.4368951916694641, 'arms_up': 0.4188764989376068, 'hair_intakes': 0.394428551197052,

            'thighs': 0.33927807211875916, 'cleft_of_venus': 0.31293246150016785, 'collarbone': 0.30162444710731506

        })

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858})

        assert rating == pytest.approx({'explicit': 0.9496212005615234})

        }, abs=1e-2)

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858}, abs=1e-2)

        assert rating == pytest.approx({'explicit': 0.9496212005615234}, abs=1e-2)



    def test_multilabel_timm_predict_with_category_thresholds_cateid_no_tagth_evalx(self):

        general, character, rating = multilabel_timm_predict(
@@ -167,6 +167,6 @@ class TestGenericMultilabelTIMM: 
            'anus': 0.5988802313804626, 'hair_between_eyes': 0.5209077596664429, 'closed_mouth': 0.5168408155441284,

            'medium_breasts': 0.4368951916694641, 'arms_up': 0.4188764989376068, 'hair_intakes': 0.394428551197052,

            'thighs': 0.33927807211875916, 'cleft_of_venus': 0.31293246150016785, 'collarbone': 0.30162444710731506

        })

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858})

        assert rating == pytest.approx({'explicit': 0.9496212005615234})
 
        }, abs=1e-2)

        assert character == pytest.approx({'surtr_(arknights)': 0.9033797979354858}, abs=1e-2)

        assert rating == pytest.approx({'explicit': 0.9496212005615234}, abs=1e-2)