added tests for migration

Author: jonas-hurst

tests/extensions/cassettes/test_mlm/test_migrate[https---raw.githubusercontent.com-stac-extensions-mlm-refs-tags-v{version}-examples-item_basic.json-1.0.0].yaml

@@ -0,0 +1,110 @@
+interactions:
+- request:
+    body: null
+    headers:
+      Accept:
+      - '*/*'
+      Accept-Encoding:
+      - gzip, deflate
+      Connection:
+      - keep-alive
+      User-Agent:
+      - python-requests/2.32.3
+    method: GET
+    uri: https://raw.githubusercontent.com/stac-extensions/mlm/refs/tags/v1.0.0/examples/item_basic.json
+  response:
+    body:
+      string: "{\n  \"stac_version\": \"1.0.0\",\n  \"stac_extensions\": [\n    \"https://crim-ca.github.io/mlm-extension/v1.0.0/schema.json\"\n
+        \ ],\n  \"type\": \"Feature\",\n  \"id\": \"example-model\",\n  \"collection\":
+        \"ml-model-examples\",\n  \"geometry\": {\n    \"type\": \"Polygon\",\n    \"coordinates\":
+        [\n      [\n        [\n          -7.882190080512502,\n          37.13739173208318\n
+        \       ],\n        [\n          -7.882190080512502,\n          58.21798141355221\n
+        \       ],\n        [\n          27.911651652899923,\n          58.21798141355221\n
+        \       ],\n        [\n          27.911651652899923,\n          37.13739173208318\n
+        \       ],\n        [\n          -7.882190080512502,\n          37.13739173208318\n
+        \       ]\n      ]\n    ]\n  },\n  \"bbox\": [\n    -7.882190080512502,\n
+        \   37.13739173208318,\n    27.911651652899923,\n    58.21798141355221\n  ],\n
+        \ \"properties\": {\n    \"description\": \"Basic STAC Item with only the
+        MLM extension and no other extension cross-references.\",\n    \"datetime\":
+        null,\n    \"start_datetime\": \"1900-01-01T00:00:00Z\",\n    \"end_datetime\":
+        \"9999-12-31T23:59:59Z\",\n    \"mlm:name\": \"example-model\",\n    \"mlm:tasks\":
+        [\n      \"classification\"\n    ],\n    \"mlm:architecture\": \"ResNet\",\n
+        \   \"mlm:input\": [\n      {\n        \"name\": \"Model with RGB input that
+        does not refer to any band.\",\n        \"bands\": [],\n        \"input\":
+        {\n          \"shape\": [\n            -1,\n            3,\n            64,\n
+        \           64\n          ],\n          \"dim_order\": [\n            \"batch\",\n
+        \           \"channel\",\n            \"height\",\n            \"width\"\n
+        \         ],\n          \"data_type\": \"float32\"\n        }\n      }\n    ],\n
+        \   \"mlm:output\": [\n      {\n        \"name\": \"classification\",\n        \"tasks\":
+        [\n          \"classification\"\n        ],\n        \"result\": {\n          \"shape\":
+        [\n            -1,\n            1\n          ],\n          \"dim_order\":
+        [\n            \"batch\",\n            \"class\"\n          ],\n          \"data_type\":
+        \"uint8\"\n        },\n        \"classification_classes\": [\n          {\n
+        \           \"value\": 0,\n            \"name\": \"BACKGROUND\",\n            \"description\":
+        \"Background non-city.\",\n            \"color_hint\": [\n              0,\n
+        \             0,\n              0\n            ]\n          },\n          {\n
+        \           \"value\": 1,\n            \"name\": \"CITY\",\n            \"description\":
+        \"A city is detected.\",\n            \"color_hint\": [\n              0,\n
+        \             0,\n              255\n            ]\n          }\n        ]\n
+        \     }\n    ]\n  },\n  \"assets\": {\n    \"model\": {\n      \"href\": \"https://huggingface.co/example/model-card\",\n
+        \     \"title\": \"Pytorch weights checkpoint\",\n      \"description\": \"Example
+        model.\",\n      \"type\": \"text/html\",\n      \"roles\": [\n        \"mlm:model\"\n
+        \     ]\n    }\n  },\n  \"links\": [\n    {\n      \"rel\": \"collection\",\n
+        \     \"href\": \"./collection.json\",\n      \"type\": \"application/json\"\n
+        \   },\n    {\n      \"rel\": \"self\",\n      \"href\": \"./item_basic.json\",\n
+        \     \"type\": \"application/geo+json\"\n    }\n  ]\n}\n"
+    headers:
+      Accept-Ranges:
+      - bytes
+      Access-Control-Allow-Origin:
+      - '*'
+      Cache-Control:
+      - max-age=300
+      Connection:
+      - keep-alive
+      Content-Encoding:
+      - gzip
+      Content-Length:
+      - '931'
+      Content-Security-Policy:
+      - default-src 'none'; style-src 'unsafe-inline'; sandbox
+      Content-Type:
+      - text/plain; charset=utf-8
+      Cross-Origin-Resource-Policy:
+      - cross-origin
+      Date:
+      - Tue, 29 Apr 2025 08:30:25 GMT
+      ETag:
+      - W/"06b55e576e4efab26350d8d7551913db9419014d18bff024af7e5818a2366f4b"
+      Expires:
+      - Tue, 29 Apr 2025 08:35:25 GMT
+      Source-Age:
+      - '120'
+      Strict-Transport-Security:
+      - max-age=31536000
+      Vary:
+      - Authorization,Accept-Encoding,Origin
+      Via:
+      - 1.1 varnish
+      X-Cache:
+      - HIT
+      X-Cache-Hits:
+      - '0'
+      X-Content-Type-Options:
+      - nosniff
+      X-Fastly-Request-ID:
+      - cbb71f87565b003068eeb6cf1d44e3e92f0b75fb
+      X-Frame-Options:
+      - deny
+      X-GitHub-Request-Id:
+      - EAA8:1C62A6:B8FA8E:D1A987:681087C4
+      X-Served-By:
+      - cache-fra-etou8220124-FRA
+      X-Timer:
+      - S1745915426.697783,VS0,VE1
+      X-XSS-Protection:
+      - 1; mode=block
+    status:
+      code: 200
+      message: OK
+version: 1

tests/extensions/cassettes/test_mlm/test_migrate[https---raw.githubusercontent.com-stac-extensions-mlm-refs-tags-v{version}-examples-item_basic.json-1.1.0].yaml

@@ -0,0 +1,110 @@
+interactions:
+- request:
+    body: null
+    headers:
+      Accept:
+      - '*/*'
+      Accept-Encoding:
+      - gzip, deflate
+      Connection:
+      - keep-alive
+      User-Agent:
+      - python-requests/2.32.3
+    method: GET
+    uri: https://raw.githubusercontent.com/stac-extensions/mlm/refs/tags/v1.1.0/examples/item_basic.json
+  response:
+    body:
+      string: "{\n  \"stac_version\": \"1.0.0\",\n  \"stac_extensions\": [\n    \"https://crim-ca.github.io/mlm-extension/v1.1.0/schema.json\"\n
+        \ ],\n  \"type\": \"Feature\",\n  \"id\": \"example-model\",\n  \"collection\":
+        \"ml-model-examples\",\n  \"geometry\": {\n    \"type\": \"Polygon\",\n    \"coordinates\":
+        [\n      [\n        [\n          -7.882190080512502,\n          37.13739173208318\n
+        \       ],\n        [\n          -7.882190080512502,\n          58.21798141355221\n
+        \       ],\n        [\n          27.911651652899923,\n          58.21798141355221\n
+        \       ],\n        [\n          27.911651652899923,\n          37.13739173208318\n
+        \       ],\n        [\n          -7.882190080512502,\n          37.13739173208318\n
+        \       ]\n      ]\n    ]\n  },\n  \"bbox\": [\n    -7.882190080512502,\n
+        \   37.13739173208318,\n    27.911651652899923,\n    58.21798141355221\n  ],\n
+        \ \"properties\": {\n    \"description\": \"Basic STAC Item with only the
+        MLM extension and no other extension cross-references.\",\n    \"datetime\":
+        null,\n    \"start_datetime\": \"1900-01-01T00:00:00Z\",\n    \"end_datetime\":
+        \"9999-12-31T23:59:59Z\",\n    \"mlm:name\": \"example-model\",\n    \"mlm:tasks\":
+        [\n      \"classification\"\n    ],\n    \"mlm:architecture\": \"ResNet\",\n
+        \   \"mlm:input\": [\n      {\n        \"name\": \"Model with RGB input that
+        does not refer to any band.\",\n        \"bands\": [],\n        \"input\":
+        {\n          \"shape\": [\n            -1,\n            3,\n            64,\n
+        \           64\n          ],\n          \"dim_order\": [\n            \"batch\",\n
+        \           \"channel\",\n            \"height\",\n            \"width\"\n
+        \         ],\n          \"data_type\": \"float32\"\n        }\n      }\n    ],\n
+        \   \"mlm:output\": [\n      {\n        \"name\": \"classification\",\n        \"tasks\":
+        [\n          \"classification\"\n        ],\n        \"result\": {\n          \"shape\":
+        [\n            -1,\n            1\n          ],\n          \"dim_order\":
+        [\n            \"batch\",\n            \"class\"\n          ],\n          \"data_type\":
+        \"uint8\"\n        },\n        \"classification_classes\": [\n          {\n
+        \           \"value\": 0,\n            \"name\": \"BACKGROUND\",\n            \"description\":
+        \"Background non-city.\",\n            \"color_hint\": [\n              0,\n
+        \             0,\n              0\n            ]\n          },\n          {\n
+        \           \"value\": 1,\n            \"name\": \"CITY\",\n            \"description\":
+        \"A city is detected.\",\n            \"color_hint\": [\n              0,\n
+        \             0,\n              255\n            ]\n          }\n        ]\n
+        \     }\n    ]\n  },\n  \"assets\": {\n    \"model\": {\n      \"href\": \"https://huggingface.co/example/model-card\",\n
+        \     \"title\": \"Pytorch weights checkpoint\",\n      \"description\": \"Example
+        model.\",\n      \"type\": \"text/html\",\n      \"roles\": [\n        \"mlm:model\"\n
+        \     ]\n    }\n  },\n  \"links\": [\n    {\n      \"rel\": \"collection\",\n
+        \     \"href\": \"./collection.json\",\n      \"type\": \"application/json\"\n
+        \   },\n    {\n      \"rel\": \"self\",\n      \"href\": \"./item_basic.json\",\n
+        \     \"type\": \"application/geo+json\"\n    }\n  ]\n}\n"
+    headers:
+      Accept-Ranges:
+      - bytes
+      Access-Control-Allow-Origin:
+      - '*'
+      Cache-Control:
+      - max-age=300
+      Connection:
+      - keep-alive
+      Content-Encoding:
+      - gzip
+      Content-Length:
+      - '933'
+      Content-Security-Policy:
+      - default-src 'none'; style-src 'unsafe-inline'; sandbox
+      Content-Type:
+      - text/plain; charset=utf-8
+      Cross-Origin-Resource-Policy:
+      - cross-origin
+      Date:
+      - Tue, 29 Apr 2025 08:30:25 GMT
+      ETag:
+      - W/"60cdf768894e2f4eaafeb01906736c2bb03903640b6615f1cea0dd4ace03a6d3"
+      Expires:
+      - Tue, 29 Apr 2025 08:35:25 GMT
+      Source-Age:
+      - '120'
+      Strict-Transport-Security:
+      - max-age=31536000
+      Vary:
+      - Authorization,Accept-Encoding,Origin
+      Via:
+      - 1.1 varnish
+      X-Cache:
+      - HIT
+      X-Cache-Hits:
+      - '0'
+      X-Content-Type-Options:
+      - nosniff
+      X-Fastly-Request-ID:
+      - fdaeae3f7f917f5f4c801f2719acdc47c172defb
+      X-Frame-Options:
+      - deny
+      X-GitHub-Request-Id:
+      - 5BEA:6EEED:C4E1E6:DE2B23:68108DA9
+      X-Served-By:
+      - cache-fra-etou8220128-FRA
+      X-Timer:
+      - S1745915426.768525,VS0,VE2
+      X-XSS-Protection:
+      - 1; mode=block
+    status:
+      code: 200
+      message: OK
+version: 1

tests/extensions/cassettes/test_mlm/test_migrate[https---raw.githubusercontent.com-stac-extensions-mlm-refs-tags-v{version}-examples-item_basic.json-1.2.0].yaml

@@ -0,0 +1,110 @@
+interactions:
+- request:
+    body: null
+    headers:
+      Accept:
+      - '*/*'
+      Accept-Encoding:
+      - gzip, deflate
+      Connection:
+      - keep-alive
+      User-Agent:
+      - python-requests/2.32.3
+    method: GET
+    uri: https://raw.githubusercontent.com/stac-extensions/mlm/refs/tags/v1.2.0/examples/item_basic.json
+  response:
+    body:
+      string: "{\n  \"stac_version\": \"1.0.0\",\n  \"stac_extensions\": [\n    \"https://crim-ca.github.io/mlm-extension/v1.2.0/schema.json\"\n
+        \ ],\n  \"type\": \"Feature\",\n  \"id\": \"example-model\",\n  \"collection\":
+        \"ml-model-examples\",\n  \"geometry\": {\n    \"type\": \"Polygon\",\n    \"coordinates\":
+        [\n      [\n        [\n          -7.882190080512502,\n          37.13739173208318\n
+        \       ],\n        [\n          -7.882190080512502,\n          58.21798141355221\n
+        \       ],\n        [\n          27.911651652899923,\n          58.21798141355221\n
+        \       ],\n        [\n          27.911651652899923,\n          37.13739173208318\n
+        \       ],\n        [\n          -7.882190080512502,\n          37.13739173208318\n
+        \       ]\n      ]\n    ]\n  },\n  \"bbox\": [\n    -7.882190080512502,\n
+        \   37.13739173208318,\n    27.911651652899923,\n    58.21798141355221\n  ],\n
+        \ \"properties\": {\n    \"description\": \"Basic STAC Item with only the
+        MLM extension and no other extension cross-references.\",\n    \"datetime\":
+        null,\n    \"start_datetime\": \"1900-01-01T00:00:00Z\",\n    \"end_datetime\":
+        \"9999-12-31T23:59:59Z\",\n    \"mlm:name\": \"example-model\",\n    \"mlm:tasks\":
+        [\n      \"classification\"\n    ],\n    \"mlm:architecture\": \"ResNet\",\n
+        \   \"mlm:input\": [\n      {\n        \"name\": \"Model with RGB input that
+        does not refer to any band.\",\n        \"bands\": [],\n        \"input\":
+        {\n          \"shape\": [\n            -1,\n            3,\n            64,\n
+        \           64\n          ],\n          \"dim_order\": [\n            \"batch\",\n
+        \           \"channel\",\n            \"height\",\n            \"width\"\n
+        \         ],\n          \"data_type\": \"float32\"\n        }\n      }\n    ],\n
+        \   \"mlm:output\": [\n      {\n        \"name\": \"classification\",\n        \"tasks\":
+        [\n          \"classification\"\n        ],\n        \"result\": {\n          \"shape\":
+        [\n            -1,\n            1\n          ],\n          \"dim_order\":
+        [\n            \"batch\",\n            \"class\"\n          ],\n          \"data_type\":
+        \"uint8\"\n        },\n        \"classification_classes\": [\n          {\n
+        \           \"value\": 0,\n            \"name\": \"BACKGROUND\",\n            \"description\":
+        \"Background non-city.\",\n            \"color_hint\": [\n              0,\n
+        \             0,\n              0\n            ]\n          },\n          {\n
+        \           \"value\": 1,\n            \"name\": \"CITY\",\n            \"description\":
+        \"A city is detected.\",\n            \"color_hint\": [\n              0,\n
+        \             0,\n              255\n            ]\n          }\n        ]\n
+        \     }\n    ]\n  },\n  \"assets\": {\n    \"model\": {\n      \"href\": \"https://huggingface.co/example/model-card\",\n
+        \     \"title\": \"Pytorch weights checkpoint\",\n      \"description\": \"Example
+        model.\",\n      \"type\": \"text/html\",\n      \"roles\": [\n        \"mlm:model\"\n
+        \     ]\n    }\n  },\n  \"links\": [\n    {\n      \"rel\": \"collection\",\n
+        \     \"href\": \"./collection.json\",\n      \"type\": \"application/json\"\n
+        \   },\n    {\n      \"rel\": \"self\",\n      \"href\": \"./item_basic.json\",\n
+        \     \"type\": \"application/geo+json\"\n    }\n  ]\n}\n"
+    headers:
+      Accept-Ranges:
+      - bytes
+      Access-Control-Allow-Origin:
+      - '*'
+      Cache-Control:
+      - max-age=300
+      Connection:
+      - keep-alive
+      Content-Encoding:
+      - gzip
+      Content-Length:
+      - '934'
+      Content-Security-Policy:
+      - default-src 'none'; style-src 'unsafe-inline'; sandbox
+      Content-Type:
+      - text/plain; charset=utf-8
+      Cross-Origin-Resource-Policy:
+      - cross-origin
+      Date:
+      - Tue, 29 Apr 2025 08:30:25 GMT
+      ETag:
+      - W/"4214e4c8a7e047eaff361f43a1e2ce58d4e403d064c4a0a9fc870a0f02f3ffd4"
+      Expires:
+      - Tue, 29 Apr 2025 08:35:25 GMT
+      Source-Age:
+      - '119'
+      Strict-Transport-Security:
+      - max-age=31536000
+      Vary:
+      - Authorization,Accept-Encoding,Origin
+      Via:
+      - 1.1 varnish
+      X-Cache:
+      - HIT
+      X-Cache-Hits:
+      - '0'
+      X-Content-Type-Options:
+      - nosniff
+      X-Fastly-Request-ID:
+      - dea14443ef6f269978dd64c2bb7cf0811dcd3102
+      X-Frame-Options:
+      - deny
+      X-GitHub-Request-Id:
+      - 6822:112C:B53704:CD19F6:68108DA9
+      X-Served-By:
+      - cache-fra-etou8220154-FRA
+      X-Timer:
+      - S1745915426.841928,VS0,VE1
+      X-XSS-Protection:
+      - 1; mode=block
+    status:
+      code: 200
+      message: OK
+version: 1