Skip to content
Snippets Groups Projects
Commit 0e7a84c2 authored by Cresson Remi's avatar Cresson Remi
Browse files

WIP: CRC book tutorial tests

parent 29d5d581
No related branches found
Tags 1.2.5
3 merge requests!15Develop,!12Crc book tests,!3Migration to TensorFlow2
Pipeline #191196 failed
...@@ -171,6 +171,35 @@ class TutorialTest(unittest.TestCase): ...@@ -171,6 +171,35 @@ class TutorialTest(unittest.TestCase):
to_compare_dict={"$DATADIR/classif_model1.tif": "$TMPDIR/classif_model1.tif"}, to_compare_dict={"$DATADIR/classif_model1.tif": "$TMPDIR/classif_model1.tif"},
tol=INFERENCE_MAE_TOL)) tol=INFERENCE_MAE_TOL))
@pytest.mark.order(7)
def test_rf_sampling(self):
self.assertTrue(
run_command_and_test_exist(command="otbcli_SampleExtraction "
"-in $DATADIR/s2_stack.jp2 "
"-vec $TMPDIR/outvec_A.gpkg "
"-field class "
"-out $TMPDIR/pixelvalues_A.gpkg",
file_list=["$TMPDIR/pixelvalues_A.gpkg"]))
self.assertTrue(
run_command_and_test_exist(command="otbcli_SampleExtraction "
"-in $DATADIR/s2_stack.jp2 "
"-vec $TMPDIR/outvec_B.gpkg "
"-field class "
"-out $TMPDIR/pixelvalues_B.gpkg",
file_list=["$TMPDIR/pixelvalues_B.gpkg"]))
@pytest.mark.order(8)
def test_rf_training(self):
self.assertTrue(
run_command_and_test_exist(command="otbcli_TrainVectorClassifier "
"-io.vd $TMPDIR/pixelvalues_A.gpkg "
"-valid.vd $TMPDIR/pixelvalues_B.gpkg "
"-feat value_0 value_1 value_2 value_3 "
"-cfield class "
"-classifier rf "
"-io.out $TMPDIR/randomforest_model.yaml ",
file_list=["$TMPDIR/randomforest_model.yaml"]))
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment