Add mock_test_generator

Author: blegat

test/mock_tests_generator.ipynb

@@ -0,0 +1,166 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "using SemidefiniteOptInterface\n",
+    "const SDOI = SemidefiniteOptInterface\n",
+    "\n",
+    "using Base.Test\n",
+    "\n",
+    "using MathOptInterface\n",
+    "const MOI = MathOptInterface\n",
+    "const MOIT = MOI.Test\n",
+    "const MOIB = MOI.Bridges\n",
+    "const MOIU = MOI.Utilities"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "We hack the `optimize!` call of CSDP to print the `mock_optimize!` function to be used in the tests for the mock_optimizer.\n",
+    "The output still needs to be edited and replaced by the exact solution but this is a good start"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "collapsed": true
+   },
+   "outputs": [],
+   "source": [
+    "const MOIU = MOI.Utilities\n",
+    "MOIU.@model SDModelData () (EqualTo, GreaterThan, LessThan) (Zeros, Nonnegatives, Nonpositives, PositiveSemidefiniteConeTriangle) () (SingleVariable,) (ScalarAffineFunction,) (VectorOfVariables,) (VectorAffineFunction,)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import CSDP\n",
+    "optimizer = CSDP.CSDPOptimizer(printlevel=0)\n",
+    "\n",
+    "cached_optimizer = MOIU.CachingOptimizer(SDModelData{Float64}(), optimizer)\n",
+    "bridged_optimizer = MOIB.SplitInterval{Float64}(MOIB.RootDet{Float64}(MOIB.GeoMean{Float64}(MOIB.RSOCtoPSD{Float64}(MOIB.SOCtoPSD{Float64}(cached_optimizer)))))\n",
+    "\n",
+    "config = MOIT.TestConfig(atol=1e-4, rtol=1e-4)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "function MOI.optimize!(m::SDOI.SOItoMOIBridge)\n",
+    "    MOI.optimize!(m.sdoptimizer)\n",
+    "    X = SDOI.getX(m.sdoptimizer)\n",
+    "    println(\"MOIU.set_mock_optimize!(mock, (mock) -> MOIU.mock_optimize!(mock,\")\n",
+    "    print(  \"                                                            \")\n",
+    "    if MOI.get(m, MOI.DualStatus()) == MOI.InfeasibilityCertificate\n",
+    "        print(\"tuple()\")\n",
+    "    else\n",
+    "        if MOI.get(m, MOI.PrimalStatus()) == MOI.InfeasibilityCertificate\n",
+    "            print(\"(MOI.InfeasibilityCertificate, \")\n",
+    "        end\n",
+    "        print(  \"[\")\n",
+    "        for (blk, dim) in enumerate(m.blockdims)\n",
+    "            if blk > 1\n",
+    "                print(\", \")\n",
+    "            end\n",
+    "            #print(round.(SDOI.block(X, blk), 4))\n",
+    "            print(SDOI.block(X, blk))\n",
+    "        end\n",
+    "        print(\"]\")\n",
+    "        if MOI.get(m, MOI.PrimalStatus()) == MOI.InfeasibilityCertificate\n",
+    "            print(\")\")\n",
+    "        end\n",
+    "    end\n",
+    "    if MOI.get(m, MOI.PrimalStatus()) != MOI.InfeasibilityCertificate\n",
+    "        println(\",\")\n",
+    "        print(  \"                                                            \")\n",
+    "        #print(round.(SDOI.gety(m.sdoptimizer), 4))\n",
+    "        print(SDOI.gety(m.sdoptimizer))\n",
+    "    end\n",
+    "    println(\"))\")\n",
+    "end"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "mock = SDOI.MockSDOptimizer{Float64}()\n",
+    "mock_optimizer = SDOI.SDOIOptimizer(mock, Float64)\n",
+    "cached_mock_optimizer = MOIU.CachingOptimizer(SDModelData{Float64}(), mock_optimizer)\n",
+    "config = MOIT.TestConfig(atol=1e-4, rtol=1e-4)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "function generate_mock_test(testfun)\n",
+    "    testfun(bridged_optimizer, config)\n",
+    "    testcall = replace(string(testfun), \"MathOptInterface.Test\", \"MOIT\")\n",
+    "    println(\"$testcall(cached_mock_optimizer, config)\")\n",
+    "end"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "generate_mock_test(MOIT.solve_singlevariable_obj)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "for testfun in map(p -> p.second, sort(collect(MOIT.psdttests), by=p->p.first))\n",
+    "    generate_mock_test(testfun)\n",
+    "end"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "collapsed": true
+   },
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Julia 0.6.3",
+   "language": "julia",
+   "name": "julia-0.6"
+  },
+  "language_info": {
+   "file_extension": ".jl",
+   "mimetype": "application/julia",
+   "name": "julia",
+   "version": "0.6.4"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}