aigroup-econ-mcp 1.3.3__py3-none-any.whl → 2.0.1__py3-none-any.whl

econometrics/tests/basic_parametric_estimation_tests/test_gmm.py

@@ -0,0 +1,128 @@
+"""
+GMM模型测试脚本
+"""
+
+import sys
+import os
+import numpy as np
+
+# 添加项目根目录到路径
+sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', '..'))
+
+from econometrics.basic_parametric_estimation.gmm.gmm_model import gmm_estimation, GMMResult
+
+
+def test_gmm_basic():
+    """测试基本GMM功能"""
+    print("测试基本GMM功能...")
+    
+    # 生成测试数据
+    np.random.seed(42)
+    n = 100
+    x1 = np.random.randn(n)
+    x2 = np.random.randn(n)
+    # 真实模型: y = 2 + 3*x1 + 2*x2 + noise
+    y = 2 + 3*x1 + 2*x2 + np.random.randn(n) * 0.5
+    
+    # 准备数据
+    x_data = [[x1[i], x2[i]] for i in range(n)]
+    y_data = y.tolist()
+    
+    # 执行GMM回归（无工具变量，退化为OLS）
+    result = gmm_estimation(y_data, x_data, feature_names=['x1', 'x2'])
+    
+    # 验证结果类型
+    assert isinstance(result, GMMResult), "结果应为GMMResult类型"
+    
+    # 验证系数数量
+    assert len(result.coefficients) == 3, "应该有3个系数（包括常数项）"
+    
+    # 验证统计量合理性
+    assert result.n_obs == n, f"观测数量应为{n}"
+    assert result.n_moments == 3, "矩条件数量应为3"
+    
+    print("  系数:", result.coefficients)
+    print("  J统计量:", result.j_statistic)
+    print("  基本GMM功能测试通过")
+
+
+def test_gmm_with_instruments():
+    """测试带工具变量的GMM"""
+    print("测试带工具变量的GMM...")
+    
+    # 生成测试数据（工具变量）
+    np.random.seed(42)
+    n = 80
+    x1 = np.random.randn(n)
+    x2 = np.random.randn(n)
+    z1 = np.random.randn(n)  # 工具变量
+    z2 = np.random.randn(n)  # 工具变量
+    # 真实模型: y = 2 + 3*x1 + 2*x2 + noise
+    y = 2 + 3*x1 + 2*x2 + np.random.randn(n) * 0.5
+    
+    # 准备数据
+    x_data = [[x1[i], x2[i]] for i in range(n)]
+    instruments = [[z1[i], z2[i]] for i in range(n)]
+    y_data = y.tolist()
+    
+    # 执行GMM回归（带工具变量）
+    result = gmm_estimation(y_data, x_data, instruments=instruments, feature_names=['x1', 'x2'])
+    
+    # 验证系数数量
+    assert len(result.coefficients) == 3, "应该有3个系数（包括常数项）"
+    
+    print("  系数:", result.coefficients)
+    print("  带工具变量的GMM测试通过")
+
+
+def test_gmm_no_constant():
+    """测试不包含常数项的GMM"""
+    print("测试不包含常数项的GMM...")
+    
+    # 生成测试数据
+    np.random.seed(42)
+    n = 50
+    x = np.random.randn(n)
+    # 真实模型: y = 2*x + noise
+    y = 2*x + np.random.randn(n) * 0.3
+    
+    # 执行GMM回归（不包含常数项）
+    result = gmm_estimation(y.tolist(), x.tolist(), constant=False, feature_names=['x1'])
+    
+    # 验证系数数量
+    assert len(result.coefficients) == 1, "应该有1个系数（不包括常数项）"
+    
+    print("  系数:", result.coefficients)
+    print("  不包含常数项的GMM测试通过")
+
+
+def test_gmm_errors():
+    """测试GMM错误处理"""
+    print("测试GMM错误处理...")
+    
+    # 测试空数据
+    try:
+        gmm_estimation([], [])
+        assert False, "应该抛出ValueError异常"
+    except ValueError:
+        print("  空数据错误处理正确")
+    
+    # 测试不一致的数据维度
+    try:
+        y_data = [1, 2, 3]
+        x_data = [[1, 2], [3, 4]]  # 维度不匹配
+        gmm_estimation(y_data, x_data)
+        assert False, "应该抛出ValueError异常"
+    except ValueError:
+        print("  数据维度不一致错误处理正确")
+    
+    print("  GMM错误处理测试通过")
+
+
+if __name__ == "__main__":
+    print("开始测试GMM模型...")
+    test_gmm_basic()
+    test_gmm_with_instruments()
+    test_gmm_no_constant()
+    test_gmm_errors()
+    print("所有GMM测试通过!")

econometrics/tests/basic_parametric_estimation_tests/test_mle.py

@@ -0,0 +1,127 @@
+"""
+MLE模型测试脚本
+"""
+
+import sys
+import os
+import numpy as np
+
+# 添加项目根目录到路径
+sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', '..'))
+
+from econometrics.basic_parametric_estimation.mle.mle_model import mle_estimation, MLEResult
+
+
+def test_mle_normal():
+    """测试正态分布MLE"""
+    print("测试正态分布MLE...")
+    
+    # 生成正态分布测试数据
+    np.random.seed(42)
+    data = np.random.normal(5, 2, 100).tolist()  # 均值5，标准差2
+    
+    # 执行MLE估计
+    result = mle_estimation(data, distribution="normal")
+    
+    # 验证结果类型
+    assert isinstance(result, MLEResult), "结果应为MLEResult类型"
+    
+    # 验证参数数量
+    assert len(result.parameters) == 2, "正态分布应有2个参数（均值和标准差）"
+    
+    # 验证参数名称
+    assert result.param_names == ["mu", "sigma"], "参数名称应为['mu', 'sigma']"
+    
+    # 验证统计量合理性
+    assert result.n_obs == 100, "观测数量应为100"
+    assert result.log_likelihood < 0, "对数似然值应为负数"
+    
+    print("  估计参数:", result.parameters)
+    print("  对数似然值:", result.log_likelihood)
+    print("  AIC:", result.aic)
+    print("  正态分布MLE测试通过")
+
+
+def test_mle_poisson():
+    """测试泊松分布MLE"""
+    print("测试泊松分布MLE...")
+    
+    # 生成泊松分布测试数据
+    np.random.seed(42)
+    data = np.random.poisson(3, 100).tolist()  # 均值3
+    
+    # 执行MLE估计
+    result = mle_estimation(data, distribution="poisson")
+    
+    # 验证结果类型
+    assert isinstance(result, MLEResult), "结果应为MLEResult类型"
+    
+    # 验证参数数量
+    assert len(result.parameters) == 1, "泊松分布应有1个参数"
+    
+    # 验证参数名称
+    assert result.param_names == ["lambda"], "参数名称应为['lambda']"
+    
+    print("  估计参数:", result.parameters)
+    print("  泊松分布MLE测试通过")
+
+
+def test_mle_exponential():
+    """测试指数分布MLE"""
+    print("测试指数分布MLE...")
+    
+    # 生成指数分布测试数据
+    np.random.seed(42)
+    data = np.random.exponential(2, 100).tolist()  # 均值2
+    
+    # 执行MLE估计
+    result = mle_estimation(data, distribution="exponential")
+    
+    # 验证结果类型
+    assert isinstance(result, MLEResult), "结果应为MLEResult类型"
+    
+    # 验证参数数量
+    assert len(result.parameters) == 1, "指数分布应有1个参数"
+    
+    # 验证参数名称
+    assert result.param_names == ["lambda"], "参数名称应为['lambda']"
+    
+    print("  估计参数:", result.parameters)
+    print("  指数分布MLE测试通过")
+
+
+def test_mle_errors():
+    """测试MLE错误处理"""
+    print("测试MLE错误处理...")
+    
+    # 测试空数据
+    try:
+        mle_estimation([])
+        assert False, "应该抛出ValueError异常"
+    except ValueError:
+        print("  空数据错误处理正确")
+    
+    # 测试不支持的分布类型
+    try:
+        mle_estimation([1, 2, 3], distribution="unsupported")
+        assert False, "应该抛出ValueError异常"
+    except ValueError:
+        print("  不支持的分布类型错误处理正确")
+    
+    # 测试负值的指数分布数据
+    try:
+        mle_estimation([-1, 0, 1], distribution="exponential")
+        assert False, "应该抛出ValueError异常"
+    except ValueError:
+        print("  负值指数分布数据错误处理正确")
+    
+    print("  MLE错误处理测试通过")
+
+
+if __name__ == "__main__":
+    print("开始测试MLE模型...")
+    test_mle_normal()
+    test_mle_poisson()
+    test_mle_exponential()
+    test_mle_errors()
+    print("所有MLE测试通过!")

econometrics/tests/basic_parametric_estimation_tests/test_ols.py

@@ -0,0 +1,100 @@
+"""
+OLS模型测试脚本
+"""
+
+import sys
+import os
+import numpy as np
+
+# 添加项目根目录到路径
+sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', '..'))
+
+from econometrics.basic_parametric_estimation.ols.ols_model import ols_regression, OLSResult
+
+
+def test_ols_basic():
+    """测试基本OLS功能"""
+    print("测试基本OLS功能...")
+    
+    # 生成测试数据
+    np.random.seed(42)
+    n = 100
+    x1 = np.random.randn(n)
+    x2 = np.random.randn(n)
+    # 真实模型: y = 2 + 3*x1 + 2*x2 + noise
+    y = 2 + 3*x1 + 2*x2 + np.random.randn(n) * 0.5
+    
+    # 准备数据
+    x_data = [[x1[i], x2[i]] for i in range(n)]
+    y_data = y.tolist()
+    
+    # 执行OLS回归
+    result = ols_regression(y_data, x_data, feature_names=['x1', 'x2'])
+    
+    # 验证结果类型
+    assert isinstance(result, OLSResult), "结果应为OLSResult类型"
+    
+    # 验证系数数量
+    assert len(result.coefficients) == 3, "应该有3个系数（包括常数项）"
+    
+    # 验证统计量合理性
+    assert 0 <= result.r_squared <= 1, "R方应在0到1之间"
+    assert result.n_obs == n, f"观测数量应为{n}"
+    
+    print("  系数:", result.coefficients)
+    print("  R方:", result.r_squared)
+    print("  调整R方:", result.adj_r_squared)
+    print("  F统计量:", result.f_statistic)
+    print("  基本OLS功能测试通过")
+
+
+def test_ols_no_constant():
+    """测试不包含常数项的OLS"""
+    print("测试不包含常数项的OLS...")
+    
+    # 生成测试数据
+    np.random.seed(42)
+    n = 50
+    x = np.random.randn(n)
+    # 真实模型: y = 2*x + noise
+    y = 2*x + np.random.randn(n) * 0.3
+    
+    # 执行OLS回归（不包含常数项）
+    result = ols_regression(y.tolist(), x.tolist(), constant=False, feature_names=['x1'])
+    
+    # 验证系数数量
+    assert len(result.coefficients) == 1, "应该有1个系数（不包括常数项）"
+    
+    print("  系数:", result.coefficients)
+    print("  不包含常数项的OLS测试通过")
+
+
+def test_ols_errors():
+    """测试OLS错误处理"""
+    print("测试OLS错误处理...")
+    
+    # 测试空数据
+    try:
+        ols_regression([], [])
+        assert False, "应该抛出ValueError异常"
+    except ValueError:
+        print("  空数据错误处理正确")
+    
+    # 测试不一致的数据维度
+    try:
+        y_data = [1, 2, 3]
+        x_data = [[1, 2], [3, 4]]  # 维度不匹配
+        ols_regression(y_data, x_data)
+        assert False, "应该抛出ValueError异常"
+    except ValueError:
+        print("  数据维度不一致错误处理正确")
+    
+    print("  OLS错误处理测试通过")
+
+
+if __name__ == "__main__":
+    print("开始测试OLS模型...")
+    test_ols_basic()
+    test_ols_no_constant()
+    test_ols_errors()
+    print("所有OLS测试通过!")

econometrics/tests/causal_inference_tests/__init__.py

@@ -0,0 +1,3 @@
+"""
+因果推断测试模块
+"""