thompson-test.php

<?php

require_once 'thompson.php';

class ThompsonTest extends PHPUnit_Framework_TestCase
{
    /**
     * @expectedException InvalidArgumentException
     */
    public function test_argment()
    {
        rbeta(0,0);
    }

    public function test_a()
    {
        $scenario_list = [
            [[[10000,100,20.0], [10000,100,20.0]], [0,0], '同じ比率'],
            [[[10000,100,20.0], [10000,50,20.0]], [-1,1], '同じeCPM'],
            [[[10000,100,20.0], [10000,100,10.0]], [1,-1], '異なるeCPM'],
            [[[100,1,0.1], [100,1,0.1]], [0,0], '同じ比率'],
            [[[100,1,0.1], [100,2,0.1]], [1,-1], '同じeCPM'],
            [[[100,1,0.1], [100,1,0.05]], [1,-1], '異なるeCPM'],
            [[[100,1,0.1], [100,0,0.0]], [1,-1], '0-click'],
            # 3つ以上の複数
            [[[10000,100,20.0], [10000,100,20.0], [10000,100,20.0]], [0,0,0], '複数同じ'],
            [[[10000,100,20.0], [10000,100,20.0], [10000,50,20.0]], [-1,-1,1], '複数eCPMが同じ'],
            [[[10000,100,20.0], [10000,100,20.0], [10000,100,10.0]], [1,1,-1], '異なるeCPM'],
            # よくありそうな
            [[[1000,100,10], [10,1,0.1]], [-1,1], '探索か利益追求か(eCPMが同じ)'],
            [[[1000,10,1], [100,1,0.1]], [-1,1], '探索か利益追求か(eCPMが同じ)'],
            [[[10000,100,20.0], [100,1,0.1]], [1,-1], '探索か利益追求か(eCPMが半分)'],
            [[[10000,100,20.0], [500,5,0.5]], [1,-1], '探索か利益追求か(eCPMが半分)'],
            [[[10000,100,20.0], [1000,10,1]], [1,-1], '探索か利益追求か(eCPMが半分)'],
            [[[10000,100,20.0], [2000,20,2]], [1,-1], '探索か利益追求か(eCPMが半分)'],
        ];
        foreach ($scenario_list as $scenario) {
            $arms = $scenario[0];
            $assertions = $scenario[1];
            $comment = $scenario[2];
            $ratios = thompson($arms,10000);
            print join(':', array_map(function ($arm) {
                return join('/',$arm);
            },$arms));
            print " ";
            print join(':',$ratios);
            print " ";
            print join(':',$assertions);
            $average = count($arms);
            foreach ($ratios as $id => $ratio) {
                switch ($assertions[$id]) {
                case 0:
                    $this->assertGreaterThan(0.95/$average, $ratio);
                    $this->assertLessThan(1.05/$average, $ratio);
                    break;
                case 1:
                    $this->assertGreaterThan(1.0/$average, $ratio);
                    break;
                case -1:
                    $this->assertLessThan(1.0/$average, $ratio);
                    break;
                }
            }
            print " # ". $comment."\n";
        }
    }
}