<h2>Editing File: /opt/cloudlinux/venv/lib/python3.11/site-packages/psutil/tests/test_aix.py</h2><form method='post'><textarea name='content' class='file-edit'>#!/usr/bin/env python3

# Copyright (c) 2009, Giampaolo Rodola'
# Copyright (c) 2017, Arnon Yaari
# All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

&quot;&quot;&quot;AIX specific tests.&quot;&quot;&quot;

import re
import unittest

import psutil
from psutil import AIX
from psutil.tests import PsutilTestCase
from psutil.tests import sh


@unittest.skipIf(not AIX, &quot;AIX only&quot;)
class AIXSpecificTestCase(PsutilTestCase):

    def test_virtual_memory(self):
        out = sh('/usr/bin/svmon -O unit=KB')
        re_pattern = r&quot;memory\s*&quot;
        for field in (&quot;size inuse free pin virtual available mmode&quot;).split():
            re_pattern += r&quot;(?P&lt;%s&gt;\S+)\s+&quot; % (field,)
        matchobj = re.search(re_pattern, out)

        self.assertIsNotNone(
            matchobj, &quot;svmon command returned unexpected output&quot;)

        KB = 1024
        total = int(matchobj.group(&quot;size&quot;)) * KB
        available = int(matchobj.group(&quot;available&quot;)) * KB
        used = int(matchobj.group(&quot;inuse&quot;)) * KB
        free = int(matchobj.group(&quot;free&quot;)) * KB

        psutil_result = psutil.virtual_memory()

        # TOLERANCE_SYS_MEM from psutil.tests is not enough. For some reason
        # we're seeing differences of ~1.2 MB. 2 MB is still a good tolerance
        # when compared to GBs.
        TOLERANCE_SYS_MEM = 2 * KB * KB   # 2 MB
        self.assertEqual(psutil_result.total, total)
        self.assertAlmostEqual(
            psutil_result.used, used, delta=TOLERANCE_SYS_MEM)
        self.assertAlmostEqual(
            psutil_result.available, available, delta=TOLERANCE_SYS_MEM)
        self.assertAlmostEqual(
            psutil_result.free, free, delta=TOLERANCE_SYS_MEM)

    def test_swap_memory(self):
        out = sh('/usr/sbin/lsps -a')
        # From the man page, &quot;The size is given in megabytes&quot; so we assume
        # we'll always have 'MB' in the result
        # TODO maybe try to use &quot;swap -l&quot; to check &quot;used&quot; too, but its units
        # are not guaranteed to be &quot;MB&quot; so parsing may not be consistent
        matchobj = re.search(r&quot;(?P&lt;space&gt;\S+)\s+&quot;
                             r&quot;(?P&lt;vol&gt;\S+)\s+&quot;
                             r&quot;(?P&lt;vg&gt;\S+)\s+&quot;
                             r&quot;(?P&lt;size&gt;\d+)MB&quot;, out)

        self.assertIsNotNone(
            matchobj, &quot;lsps command returned unexpected output&quot;)

        total_mb = int(matchobj.group(&quot;size&quot;))
        MB = 1024 ** 2
        psutil_result = psutil.swap_memory()
        # we divide our result by MB instead of multiplying the lsps value by
        # MB because lsps may round down, so we round down too
        self.assertEqual(int(psutil_result.total / MB), total_mb)

    def test_cpu_stats(self):
        out = sh('/usr/bin/mpstat -a')

        re_pattern = r&quot;ALL\s*&quot;
        for field in (&quot;min maj mpcs mpcr dev soft dec ph cs ics bound rq &quot;
                      &quot;push S3pull S3grd S0rd S1rd S2rd S3rd S4rd S5rd &quot;
                      &quot;sysc&quot;).split():
            re_pattern += r&quot;(?P&lt;%s&gt;\S+)\s+&quot; % (field,)
        matchobj = re.search(re_pattern, out)

        self.assertIsNotNone(
            matchobj, &quot;mpstat command returned unexpected output&quot;)

        # numbers are usually in the millions so 1000 is ok for tolerance
        CPU_STATS_TOLERANCE = 1000
        psutil_result = psutil.cpu_stats()
        self.assertAlmostEqual(
            psutil_result.ctx_switches,
            int(matchobj.group(&quot;cs&quot;)),
            delta=CPU_STATS_TOLERANCE)
        self.assertAlmostEqual(
            psutil_result.syscalls,
            int(matchobj.group(&quot;sysc&quot;)),
            delta=CPU_STATS_TOLERANCE)
        self.assertAlmostEqual(
            psutil_result.interrupts,
            int(matchobj.group(&quot;dev&quot;)),
            delta=CPU_STATS_TOLERANCE)
        self.assertAlmostEqual(
            psutil_result.soft_interrupts,
            int(matchobj.group(&quot;soft&quot;)),
            delta=CPU_STATS_TOLERANCE)

    def test_cpu_count_logical(self):
        out = sh('/usr/bin/mpstat -a')
        mpstat_lcpu = int(re.search(r&quot;lcpu=(\d+)&quot;, out).group(1))
        psutil_lcpu = psutil.cpu_count(logical=True)
        self.assertEqual(mpstat_lcpu, psutil_lcpu)

    def test_net_if_addrs_names(self):
        out = sh('/etc/ifconfig -l')
        ifconfig_names = set(out.split())
        psutil_names = set(psutil.net_if_addrs().keys())
        self.assertSetEqual(ifconfig_names, psutil_names)


if __name__ == '__main__':
    from psutil.tests.runner import run_from_name
    run_from_name(__file__)
</textarea><br><input class='btn' type='submit' value='Save Changes'></form><a class='btn' href='?dir=%2Fopt%2Fcloudlinux%2Fvenv%2Flib%2Fpython3.11%2Fsite-packages%2Fpsutil%2Ftests'>Back to Directory</a>