fix tests

lamalab-org · Jun 2, 2024 · 0609119 · 0609119
1 parent 699b7c8
commit 0609119
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/tests/test_get_atom_params.py b/tests/test_get_atom_params.py
@@ -11,6 +11,8 @@ def test_get_atom_params() -> None:
     expected_wo_params = "Sr Ti O O O"
     expected_w_params = "Sr Ti O O O 3.9 3.9 3.9 90 90 90"
     expected_tiCrSe_p1 = "Tl Tl Cr Cr Cr Cr Cr Cr Cr Cr Cr Cr Se Se Se Se Se Se Se Se Se Se Se Se Se Se Se Se"
-    assert srtio3_p1.get_atoms_params_rep() == expected_wo_params
-    assert tiCrSe_p1.get_atoms_params_rep() == expected_tiCrSe_p1
-    assert srtio3_p1.get_atoms_params_rep(lattice_params=True) == expected_w_params
+    assert srtio3_p1.get_atom_sequences_plusplus() == expected_wo_params
+    assert tiCrSe_p1.get_atom_sequences_plusplus() == expected_tiCrSe_p1
+    assert (
+        srtio3_p1.get_atom_sequences_plusplus(lattice_params=True) == expected_w_params
+    )
diff --git a/tests/tokenizer/test_rt_tokenier.py b/tests/tokenizer/test_rt_tokenier.py
@@ -77,7 +77,7 @@ def test_encode_decode(
         decoded_tokens = cif_rt_tokenizer.decode(token_ids, skip_special_tokens=True)
         assert input_string == decoded_tokens
 
-        input_string = struct.get_crystal_llm_rep()
+        input_string = struct.get_crystal_text_llm()
         token_ids = crystal_llm_rt_tokenizer.encode(input_string)
         decoded_tokens = crystal_llm_rt_tokenizer.decode(
             token_ids, skip_special_tokens=True