add training test file

Author: jshuadvd

tests/test_main.py

@@ -0,0 +1,11 @@
+import pytest
+import torch
+from transformers import GPT2Tokenizer
+from src.main import (
+    LongRoPEModel,
+    load_data,
+    non_uniform_interpolation,
+    RoPEPositionalEncoding,
+    progressive_extension,
+    short_context_recovery,
+)