improve coverage

Author: almogdepaz

chia/_tests/plotting/test_prover.py

@@ -75,6 +75,14 @@ def test_unsupported_file_extension_raises_value_error(self) -> None:
             get_prover_from_file("/nonexistent/path/test.txt")
 
 
+class TestV1Prover:
+    def test_v1_prover_get_version(self) -> None:
+        """Test that V1Prover.get_version() returns PlotVersion.V1"""
+        mock_disk_prover = MagicMock()
+        prover = V1Prover(mock_disk_prover)
+        assert prover.get_version() == PlotVersion.V1
+
+
 class TestGetProverFromBytes:
     def test_get_prover_from_bytes_v2_plot(self) -> None:
         with patch("chia.plotting.prover.V2Prover.from_bytes") as mock_v2_from_bytes:

chia/plotting/prover.py

@@ -120,14 +120,7 @@ def get_full_proof(self, challenge: bytes, index: int, parallel_read: bool = Tru
 
     @classmethod
     def from_bytes(cls, data: bytes) -> V1Prover:
-        from chiapos import DiskProver
-
-        disk_prover = DiskProver.from_bytes(data)
-        return cls(disk_prover)
-
-    @property
-    def disk_prover(self) -> DiskProver:
-        return self._disk_prover
+        return cls(DiskProver.from_bytes(data))
 
 
 def get_prover_from_bytes(filename: str, prover_data: bytes) -> ProverProtocol: