From 11cfa8d1febafd1c6b5f88547e5f71ffb65a8385 Mon Sep 17 00:00:00 2001
From: Gabriele <73099233+GabrielePalazzo@users.noreply.github.com>
Date: Sat, 23 Mar 2024 19:52:17 +0400
Subject: [PATCH] test: Updated tests

---
 tests/test_instruments_bluefors.py | 22 +++-------------------
 1 file changed, 3 insertions(+), 19 deletions(-)

diff --git a/tests/test_instruments_bluefors.py b/tests/test_instruments_bluefors.py
index ab08fa4f7..84975b0cd 100644
--- a/tests/test_instruments_bluefors.py
+++ b/tests/test_instruments_bluefors.py
@@ -1,7 +1,7 @@
-import datetime
 from unittest import mock
 
 import pytest
+import yaml
 
 from qibolab.instruments.bluefors import TemperatureController
 
@@ -12,20 +12,6 @@
 ]
 
 
-@pytest.mark.parametrize("message", messages)
-def test_message_converted_json(message):
-    tc = TemperatureController("Test_Temperature_Controller", "")
-    converted_message = tc.convert_to_json(message)
-    assert isinstance(converted_message, dict)
-    for key, value in converted_message.items():
-        assert key.endswith("-flange")
-        assert isinstance(value, dict)
-        assert set(value.keys()) == {"temperature", "timestamp", "time"}
-        assert isinstance(value["temperature"], float)
-        assert isinstance(value["timestamp"], float)
-        assert isinstance(value["time"], datetime.datetime)
-
-
 def test_connect():
     with mock.patch("socket.socket"):
         tc = TemperatureController("Test_Temperature_Controller", "")
@@ -51,12 +37,10 @@ def test_disconnect(already_connected):
 def test_continuously_read_data():
     with mock.patch(
         "qibolab.instruments.bluefors.TemperatureController.get_data",
-        new=lambda _: TemperatureController.convert_to_json(messages[0]),
+        new=lambda _: yaml.safe_load(messages[0]),
     ):
         tc = TemperatureController("Test_Temperature_Controller", "")
         read_temperatures = tc.read_data()
         for read_temperature in read_temperatures:
-            assert read_temperature == TemperatureController.convert_to_json(
-                messages[0]
-            )
+            assert read_temperature == yaml.safe_load(messages[0])
             break