Coverage for tests/netlistsvg/netlist_json_test.py: 100%

1# 

2# -------------------------------------------------------------------------------- 

3# SPDX-FileCopyrightText: 2024 Martin Jan Köhler and Harald Pretl 

4# Johannes Kepler University, Institute for Integrated Circuits. 

5# 

6# This file is part of KPEX  

7# (see https://github.com/martinjankoehler/klayout-pex). 

8# 

9# This program is free software: you can redistribute it and/or modify 

10# it under the terms of the GNU General Public License as published by 

11# the Free Software Foundation, either version 3 of the License, or 

12# (at your option) any later version. 

13# 

14# This program is distributed in the hope that it will be useful, 

15# but WITHOUT ANY WARRANTY; without even the implied warranty of 

16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

17# GNU General Public License for more details. 

18# 

19# You should have received a copy of the GNU General Public License 

20# along with this program. If not, see <http://www.gnu.org/licenses/>. 

21# SPDX-License-Identifier: GPL-3.0-or-later 

22# -------------------------------------------------------------------------------- 

23# 

24import allure 

25import json 

26import os 

27import tempfile 

28from typing import * 

29import unittest 

30 

31import klayout.db as kdb 

32 

33from klayout_pex.log import ( 

34 LogLevel, 

35 set_log_level, 

36) 

37from klayout_pex.netlistsvg.netlist_json import NetlistJSONWriter 

38 

39 

40@allure.parent_suite("Unit Tests") 

41@allure.tag("Netlist", "Netlist SVG Diagram") 

42class Test(unittest.TestCase): 

43 @classmethod 

44 def setUpClass(cls): 

45 set_log_level(LogLevel.DEBUG) 

46 

47 @property 

48 def netlistsvg_testdata_dir(self) -> str: 

49 return os.path.realpath(os.path.join(__file__, '..', '..', '..', 

50 'testdata', 'netlistsvg', 'netlists')) 

51 

52 def _test_netlistsvg(self, netlist_path: str, cell_name: str) -> Dict[str, Any]: 

53 netlist = kdb.Netlist() 

54 reader = kdb.NetlistSpiceReader() 

55 netlist.read(netlist_path, reader) 

56 

57 writer = NetlistJSONWriter() 

58 

59 circuit = netlist.circuit_by_name(cell_name) 

60 dict = writer.netlist_json_dict(netlist=netlist, top_circuit=circuit) 

61 

62 out_path = tempfile.mktemp(prefix=f"{cell_name}_netlistsvg_input_file", suffix=".json") 

63 writer.write_json(netlist=netlist, top_circuit=circuit, output_path=out_path) 

64 

65 print(f"Wrote reduced netlist to: {out_path}") 

66 allure.attach.file(out_path, attachment_type=allure.attachment_type.TEXT) 

67 return dict 

68 

69 def test_netlist_reduction_1(self): 

70 netlist_path = os.path.join(self.netlistsvg_testdata_dir, 'r_wire_voltage_divider_li1.pex.spice') 

71 d = self._test_netlistsvg(netlist_path=netlist_path, cell_name='r_wire_voltage_divider_li1') 

72 print(json.dumps(d, indent=4)) 

73 

74 def test_netlist_reduction_2(self): 

75 netlist_path = os.path.join(self.netlistsvg_testdata_dir, 'sky130_fd_sc_hd__inv_1.pex.spice') 

76 d = self._test_netlistsvg(netlist_path=netlist_path, cell_name='sky130_fd_sc_hd__inv_1') 

77 print(json.dumps(d, indent=4))