[feat] rename df to biopandas_structure

Author: kierandidi

biopandas/mmcif/tests/test_multiple_models.py

@@ -11,21 +11,20 @@
 TESTDATA_FILENAME = os.path.join(os.path.dirname(__file__), "data", "2jyf.cif.gz")
 
 def test_label_models():
-    df = PandasMmcif().read_mmcif(TESTDATA_FILENAME)
-    df.label_models()
-    assert "model_id" in df.df["ATOM"].columns
-    
+    biopandas_structure = PandasMmcif().read_mmcif(TESTDATA_FILENAME)
+    biopandas_structure.label_models()
+    assert "model_id" in biopandas_structure.df["ATOM"].columns
+
 def test_get_model():
-    df = PandasMmcif().read_mmcif(TESTDATA_FILENAME)
+    biopandas_structure = PandasMmcif().read_mmcif(TESTDATA_FILENAME)
     MODEL_INDEX = 1
-    new_df = df.get_model(MODEL_INDEX)
-    print(df)
-    assert new_df.df["ATOM"]["pdbx_PDB_model_num"].all() == MODEL_INDEX
+    new_biopandas_structure = biopandas_structure.get_model(MODEL_INDEX)
+    assert new_biopandas_structure.df["ATOM"]["pdbx_PDB_model_num"].all() == MODEL_INDEX
 
 
 def test_get_models():
-    df = PandasMmcif().read_mmcif(TESTDATA_FILENAME)
+    biopandas_structure = PandasMmcif().read_mmcif(TESTDATA_FILENAME)
     MODEL_INDICES = [1, 3, 5]
 
-    new_df = df.get_models(MODEL_INDICES)
-    assert new_df.df["ATOM"]["pdbx_PDB_model_num"].all() in MODEL_INDICES
+    new_biopandas_structure = biopandas_structure.get_models(MODEL_INDICES)
+    assert new_biopandas_structure.df["ATOM"]["pdbx_PDB_model_num"].all() in MODEL_INDICES