Fixed PSU UTs.

Signed-off-by: Nazarii Hnydyn <nazariig@mellanox.com>

src/sonic_ax_impl/mibs/vendor/cisco/ciscoEntityFruControlMIB.py

@@ -1,11 +1,6 @@
-import imp
-import re
-import sys
-
 from enum import Enum, unique
 from sonic_ax_impl import mibs
-from ax_interface import MIBMeta, ValueType, MIBUpdater, MIBEntry, SubtreeMIBEntry
-from ax_interface.encodings import ObjectIdentifier
+from ax_interface import MIBMeta, ValueType, SubtreeMIBEntry
 from swsssdk import SonicV2Connector
 
 CHASSIS_INFO_KEY_TEMPLATE = 'chassis {}'

tests/__init__.py

@@ -1 +0,0 @@
-import tests.mock_tables.imp

tests/mock_tables/state_db.json

@@ -2,28 +2,43 @@
   "DEVICE_METADATA|localhost": {
     "chassis_serial_number": "SAMPLETESTSN"
   },
+  "PSU_INFO|PSU 1": {
+    "presence": "false",
+    "status": "false"
+  },
+  "PSU_INFO|PSU 2": {
+    "presence": "true",
+    "status": "true"
+  },
+  "PSU_INFO|PSU 3": {
+    "presence": "true",
+    "status": "false"
+  },
+  "CHASSIS_INFO|chassis 1": {
+    "psu_num": "3"
+  },
   "TRANSCEIVER_INFO|Ethernet0": {
-  	"type": "QSFP+",
-  	"hardwarerev" : "A1",
-  	"serialnum": "SERIAL_NUM",
-  	"manufacturename": "VENDOR_NAME",
-  	"modelname": "MODEL_NAME"
+    "type": "QSFP+",
+    "hardwarerev" : "A1",
+    "serialnum": "SERIAL_NUM",
+    "manufacturename": "VENDOR_NAME",
+    "modelname": "MODEL_NAME"
   },
   "TRANSCEIVER_DOM_SENSOR|Ethernet0": {
-  	"temperature": 25.39,
-  	"voltage": 3.37,
-  	"tx1bias": "N/A",
-  	"tx2bias": 4.44,
-  	"tx3bias": "inf",
-  	"tx4bias": 4.44,
-  	"rx1power": "-inf",
-  	"rx2power": -0.97,
-  	"rx3power": -0.97,
-  	"rx4power": -0.97,
-  	"tx1power": -5.4,
-  	"tx2power": -5.4,
-  	"tx3power": -5.4,
-  	"tx4power": -5.4
+    "temperature": 25.39,
+    "voltage": 3.37,
+    "tx1bias": "N/A",
+    "tx2bias": 4.44,
+    "tx3bias": "inf",
+    "tx4bias": 4.44,
+    "rx1power": "-inf",
+    "rx2power": -0.97,
+    "rx3power": -0.97,
+    "rx4power": -0.97,
+    "tx1power": -5.4,
+    "tx2power": -5.4,
+    "tx3power": -5.4,
+    "tx4power": -5.4
   },
   "MGMT_PORT_TABLE|eth0": {
     "oper_status": "down"