python 测试磁盘读写速度和内存读写速度.

import time 
import os
# from SpeedTest import perform_default_speedtest

# 测试硬盘读写速度
def test_disk_speed():
    filename = "./testfile.bin"
    # 生成一个1GB大小的测试文件
    with open(filename, "wb") as f:
        f.write(os.urandom(1024*1024*1024))
    # 从磁盘读取文件并记录时间
    start_time = time.time()
    with open(filename, "rb") as f:
        while f.read(1024*1024):
            pass
    end_time = time.time()
    # 计算磁盘读取速度
    speed = os.path.getsize(filename) / (end_time - start_time) / (1024*1024)
    print("磁盘读取速度: %.2f MB/s" % speed)
     # 删除测试文件
    os.remove(filename)

# 测试内存读写速度
def test_memory_speed():
    # 生成一个1GB大小的测试数据
    data = bytearray(os.urandom(1024*1024*1024))
    # 从内存读取数据并记录时间
    start_time = time.time()
    for i in range(1024):
        chunk = data[i*1024*1024:(i+1)*1024*1024]
    end_time = time.time()
    # 计算内存读取速度
    speed = len(data) / (end_time - start_time) / (1024*1024)
    print("内存读取速度: %.2f MB/s" % speed)

# 测试网络速度
# def test_network_speed():
    # 创建 SpeedTest 实例
    # speedtest = SpeedTest("www.speedtest.cn", debug=1, runs=2)
    # perform_detault_speedtest("www.speedtest.cn")
    # st = SpeedTest()

    # # 测试下载速度
    # download_speed = st.download() / (1024**2)  # 将结果转换成 MB/s
    # print("下载速度: %.2f MB/s" % download_speed)

    # # 测试上传速度
    # upload_speed = st.upload() / (1024**2)  # 将结果转换成 MB/s
    # print("上传速度: %.2f MB/s" % upload_speed)

    # # 测试延迟
    # ping = st.ping()
    # print("延迟: %.2f ms" % ping)

if __name__ == "__main__":
    # test_network_speed()
    test_disk_speed()
    test_memory_speed()

下面是 SpeedTest .py文件内容. 用来测试网络速度的. 但是目前似乎有问题. 应该是规则变了 .
后来改用了一个linux的命令行工具进行的网速测试.

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from __future__ import print_function

import argparse
import bisect
import itertools
import logging
import random
import re
import string
import sys
import platform

from math import sqrt
from threading import currentThread, Thread
from time import time

try:
    from httplib import HTTPConnection
except ImportError:
    from http.client import HTTPConnection

try:
    from urllib import urlencode
except ImportError:
    from urllib.parse import urlencode

__program__ = 'pyspeedtest'
__version__ = '1.2.7'
__description__ = 'Test your bandwidth speed using Speedtest.net servers.'

__supported_formats__ = ('default', 'json', 'xml')


class SpeedTest(object):

    USER_AGENTS = {
   
   
        'Linux': 'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:41.0) Gecko/20100101 Firefox/41.0',
        'Darwin': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10.10; rv:41.0) Gecko/20100101 Firefox/41.0',
        'Windows': 'Mozilla/5.0 (Windows NT 6.3; WOW64; rv:41.0) Gecko/20100101 Firefox/41.0',
        'Java': 'Java/1.6.0_12',
    }

    DOWNLOAD_FILES = [
        '/speedtest/random350x350.jpg',
        '/speedtest/random500x500.jpg',
        '/speedtest/random1500x1500.jpg'
    ]

    UPLOAD_FILES = [
        132884,
        493638
    ]

    ALPHABET = string.digits + string.ascii_letters

    def __init__(self, host=None, http_debug=0, runs=2):
        self._host = host
        self.http_debug = http_debug
        self.runs = runs

    @property
    def host(self):
        if not self._host:
            self._host = self.chooseserver()
        return self._host

    @host.setter
    def host(self, new_host):
        self._host = new_host

    def connect(self, url):
        try:
            connection = HTTPConnection(url)
            connection.set_debuglevel(self.http_debug)
            connection.connect()
            return connection
        except:
            raise Exception('Unable to connect to %r' % url)

    def downloadthread(self, connection