oops, deleted too many tests.

Author: alxmrs

tests/test_encoder.py

@@ -2,7 +2,7 @@
 from perceiver_pytorch import Perceiver
 
 from ocean_emulators.constants import Lat, Lon
-from ocean_emulators.models.modules.encoder import PerceiverEncoder
+from ocean_emulators.models.modules.encoder import PerceiverEncoder, patch_from
 
 LATENT_DIM = 4
 
@@ -84,3 +84,21 @@ def test_makes_patches__more_variables():
     patches = encoder(prog, boundary, make_resolution(prog))
 
     assert patches.shape == (1, embed_dim, 1, 2)
+
+
+def test_patch_from__full_globe():
+    # Full globe extent should equal grid dimensions
+    patch_h, patch_w = patch_from(
+        patch_extent=(180.0, 360.0), input_height=4, input_width=8
+    )
+    assert patch_h == 4
+    assert patch_w == 8
+
+
+def test_patch_from__half_extent():
+    # Half the extent should give half the patch size
+    patch_h, patch_w = patch_from(
+        patch_extent=(90.0, 180.0), input_height=4, input_width=8
+    )
+    assert patch_h == 2
+    assert patch_w == 4