libgo/go/compress/flate/deflate_test.go

   1 // Copyright 2009 The Go Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style
   3 // license that can be found in the LICENSE file.
   4
   5 package flate
   6
   7 import (
   8         "bytes"
   9         "fmt"
  10         "io/ioutil"
  11         "os"
  12         "testing"
  13 )
  14
  15 type deflateTest struct {
  16         in    []byte
  17         level int
  18         out   []byte
  19 }
  20
  21 type deflateInflateTest struct {
  22         in []byte
  23 }
  24
  25 type reverseBitsTest struct {
  26         in       uint16
  27         bitCount uint8
  28         out      uint16
  29 }
  30
  31 var deflateTests = []*deflateTest{
  32         &deflateTest{[]byte{}, 0, []byte{1, 0, 0, 255, 255}},
  33         &deflateTest{[]byte{0x11}, -1, []byte{18, 4, 4, 0, 0, 255, 255}},
  34         &deflateTest{[]byte{0x11}, DefaultCompression, []byte{18, 4, 4, 0, 0, 255, 255}},
  35         &deflateTest{[]byte{0x11}, 4, []byte{18, 4, 4, 0, 0, 255, 255}},
  36
  37         &deflateTest{[]byte{0x11}, 0, []byte{0, 1, 0, 254, 255, 17, 1, 0, 0, 255, 255}},
  38         &deflateTest{[]byte{0x11, 0x12}, 0, []byte{0, 2, 0, 253, 255, 17, 18, 1, 0, 0, 255, 255}},
  39         &deflateTest{[]byte{0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11}, 0,
  40                 []byte{0, 8, 0, 247, 255, 17, 17, 17, 17, 17, 17, 17, 17, 1, 0, 0, 255, 255},
  41         },
  42         &deflateTest{[]byte{}, 1, []byte{1, 0, 0, 255, 255}},
  43         &deflateTest{[]byte{0x11}, 1, []byte{18, 4, 4, 0, 0, 255, 255}},
  44         &deflateTest{[]byte{0x11, 0x12}, 1, []byte{18, 20, 2, 4, 0, 0, 255, 255}},
  45         &deflateTest{[]byte{0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11}, 1, []byte{18, 132, 2, 64, 0, 0, 0, 255, 255}},
  46         &deflateTest{[]byte{}, 9, []byte{1, 0, 0, 255, 255}},
  47         &deflateTest{[]byte{0x11}, 9, []byte{18, 4, 4, 0, 0, 255, 255}},
  48         &deflateTest{[]byte{0x11, 0x12}, 9, []byte{18, 20, 2, 4, 0, 0, 255, 255}},
  49         &deflateTest{[]byte{0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11}, 9, []byte{18, 132, 2, 64, 0, 0, 0, 255, 255}},
  50 }
  51
  52 var deflateInflateTests = []*deflateInflateTest{
  53         &deflateInflateTest{[]byte{}},
  54         &deflateInflateTest{[]byte{0x11}},
  55         &deflateInflateTest{[]byte{0x11, 0x12}},
  56         &deflateInflateTest{[]byte{0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11}},
  57         &deflateInflateTest{[]byte{0x11, 0x10, 0x13, 0x41, 0x21, 0x21, 0x41, 0x13, 0x87, 0x78, 0x13}},
  58         &deflateInflateTest{getLargeDataChunk()},
  59 }
  60
  61 var reverseBitsTests = []*reverseBitsTest{
  62         &reverseBitsTest{1, 1, 1},
  63         &reverseBitsTest{1, 2, 2},
  64         &reverseBitsTest{1, 3, 4},
  65         &reverseBitsTest{1, 4, 8},
  66         &reverseBitsTest{1, 5, 16},
  67         &reverseBitsTest{17, 5, 17},
  68         &reverseBitsTest{257, 9, 257},
  69         &reverseBitsTest{29, 5, 23},
  70 }
  71
  72 func getLargeDataChunk() []byte {
  73         result := make([]byte, 100000)
  74         for i := range result {
  75                 result[i] = byte(int64(i) * int64(i) & 0xFF)
  76         }
  77         return result
  78 }
  79
  80 func TestDeflate(t *testing.T) {
  81         for _, h := range deflateTests {
  82                 buffer := bytes.NewBuffer([]byte{})
  83                 w := NewWriter(buffer, h.level)
  84                 w.Write(h.in)
  85                 w.Close()
  86                 if bytes.Compare(buffer.Bytes(), h.out) != 0 {
  87                         t.Errorf("buffer is wrong; level = %v, buffer.Bytes() = %v, expected output = %v",
  88                                 h.level, buffer.Bytes(), h.out)
  89                 }
  90         }
  91 }
  92
  93 func testToFromWithLevel(t *testing.T, level int, input []byte, name string) os.Error {
  94         buffer := bytes.NewBuffer([]byte{})
  95         w := NewWriter(buffer, level)
  96         w.Write(input)
  97         w.Close()
  98         decompressor := NewReader(buffer)
  99         decompressed, err := ioutil.ReadAll(decompressor)
 100         if err != nil {
 101                 t.Errorf("reading decompressor: %s", err)
 102                 return err
 103         }
 104         decompressor.Close()
 105         if bytes.Compare(input, decompressed) != 0 {
 106                 t.Errorf("decompress(compress(data)) != data: level=%d input=%s", level, name)
 107         }
 108         return nil
 109 }
 110
 111 func testToFrom(t *testing.T, input []byte, name string) {
 112         for i := 0; i < 10; i++ {
 113                 testToFromWithLevel(t, i, input, name)
 114         }
 115 }
 116
 117 func TestDeflateInflate(t *testing.T) {
 118         for i, h := range deflateInflateTests {
 119                 testToFrom(t, h.in, fmt.Sprintf("#%d", i))
 120         }
 121 }
 122
 123 func TestReverseBits(t *testing.T) {
 124         for _, h := range reverseBitsTests {
 125                 if v := reverseBits(h.in, h.bitCount); v != h.out {
 126                         t.Errorf("reverseBits(%v,%v) = %v, want %v",
 127                                 h.in, h.bitCount, v, h.out)
 128                 }
 129         }
 130 }
 131
 132 func TestDeflateInflateString(t *testing.T) {
 133         gold := bytes.NewBufferString(getEdata()).Bytes()
 134         testToFromWithLevel(t, 1, gold, "2.718281828...")
 135 }
 136
 137 func getEdata() string {
 138         return "2.718281828459045235360287471352662497757247093699959574966967627724076630353547" +
 139                 "59457138217852516642742746639193200305992181741359662904357290033429526059563073" +
 140                 "81323286279434907632338298807531952510190115738341879307021540891499348841675092" +
 141                 "44761460668082264800168477411853742345442437107539077744992069551702761838606261" +
 142                 "33138458300075204493382656029760673711320070932870912744374704723069697720931014" +
 143                 "16928368190255151086574637721112523897844250569536967707854499699679468644549059" +
 144                 "87931636889230098793127736178215424999229576351482208269895193668033182528869398" +
 145                 "49646510582093923982948879332036250944311730123819706841614039701983767932068328" +
 146                 "23764648042953118023287825098194558153017567173613320698112509961818815930416903" +
 147                 "51598888519345807273866738589422879228499892086805825749279610484198444363463244" +
 148                 "96848756023362482704197862320900216099023530436994184914631409343173814364054625" +
 149                 "31520961836908887070167683964243781405927145635490613031072085103837505101157477" +
 150                 "04171898610687396965521267154688957035035402123407849819334321068170121005627880" +
 151                 "23519303322474501585390473041995777709350366041699732972508868769664035557071622" +
 152                 "68447162560798826517871341951246652010305921236677194325278675398558944896970964" +
 153                 "09754591856956380236370162112047742722836489613422516445078182442352948636372141" +
 154                 "74023889344124796357437026375529444833799801612549227850925778256209262264832627" +
 155                 "79333865664816277251640191059004916449982893150566047258027786318641551956532442" +
 156                 "58698294695930801915298721172556347546396447910145904090586298496791287406870504" +
 157                 "89585867174798546677575732056812884592054133405392200011378630094556068816674001" +
 158                 "69842055804033637953764520304024322566135278369511778838638744396625322498506549" +
 159                 "95886234281899707733276171783928034946501434558897071942586398772754710962953741" +
 160                 "52111513683506275260232648472870392076431005958411661205452970302364725492966693" +
 161                 "81151373227536450988890313602057248176585118063036442812314965507047510254465011" +
 162                 "72721155519486685080036853228183152196003735625279449515828418829478761085263981" +
 163                 "39559900673764829224437528718462457803619298197139914756448826260390338144182326" +
 164                 "25150974827987779964373089970388867782271383605772978824125611907176639465070633" +
 165                 "04527954661855096666185664709711344474016070462621568071748187784437143698821855" +
 166                 "96709591025968620023537185887485696522000503117343920732113908032936344797273559" +
 167                 "55277349071783793421637012050054513263835440001863239914907054797780566978533580" +
 168                 "48966906295119432473099587655236812859041383241160722602998330535370876138939639" +
 169                 "17795745401613722361878936526053815584158718692553860616477983402543512843961294" +
 170                 "60352913325942794904337299085731580290958631382683291477116396337092400316894586" +
 171                 "36060645845925126994655724839186564209752685082307544254599376917041977780085362" +
 172                 "73094171016343490769642372229435236612557250881477922315197477806056967253801718" +
 173                 "07763603462459278778465850656050780844211529697521890874019660906651803516501792" +
 174                 "50461950136658543663271254963990854914420001457476081930221206602433009641270489" +
 175                 "43903971771951806990869986066365832322787093765022601492910115171776359446020232" +
 176                 "49300280401867723910288097866605651183260043688508817157238669842242201024950551" +
 177                 "88169480322100251542649463981287367765892768816359831247788652014117411091360116" +
 178                 "49950766290779436460058519419985601626479076153210387275571269925182756879893027" +
 179                 "61761146162549356495903798045838182323368612016243736569846703785853305275833337" +
 180                 "93990752166069238053369887956513728559388349989470741618155012539706464817194670" +
 181                 "83481972144888987906765037959036696724949925452790337296361626589760394985767413" +
 182                 "97359441023744329709355477982629614591442936451428617158587339746791897571211956" +
 183                 "18738578364475844842355558105002561149239151889309946342841393608038309166281881" +
 184                 "15037152849670597416256282360921680751501777253874025642534708790891372917228286" +
 185                 "11515915683725241630772254406337875931059826760944203261924285317018781772960235" +
 186                 "41306067213604600038966109364709514141718577701418060644363681546444005331608778" +
 187                 "31431744408119494229755993140118886833148328027065538330046932901157441475631399" +
 188                 "97221703804617092894579096271662260740718749975359212756084414737823303270330168" +
 189                 "23719364800217328573493594756433412994302485023573221459784328264142168487872167" +
 190                 "33670106150942434569844018733128101079451272237378861260581656680537143961278887" +
 191                 "32527373890392890506865324138062796025930387727697783792868409325365880733988457" +
 192                 "21874602100531148335132385004782716937621800490479559795929059165547050577751430" +
 193                 "81751126989851884087185640260353055837378324229241856256442550226721559802740126" +
 194                 "17971928047139600689163828665277009752767069777036439260224372841840883251848770" +
 195                 "47263844037953016690546593746161932384036389313136432713768884102681121989127522" +
 196                 "30562567562547017250863497653672886059667527408686274079128565769963137897530346" +
 197                 "60616669804218267724560530660773899624218340859882071864682623215080288286359746" +
 198                 "83965435885668550377313129658797581050121491620765676995065971534476347032085321" +
 199                 "56036748286083786568030730626576334697742956346437167093971930608769634953288468" +
 200                 "33613038829431040800296873869117066666146800015121143442256023874474325250769387" +
 201                 "07777519329994213727721125884360871583483562696166198057252661220679754062106208" +
 202                 "06498829184543953015299820925030054982570433905535701686531205264956148572492573" +
 203                 "86206917403695213533732531666345466588597286659451136441370331393672118569553952" +
 204                 "10845840724432383558606310680696492485123263269951460359603729725319836842336390" +
 205                 "46321367101161928217111502828016044880588023820319814930963695967358327420249882" +
 206                 "45684941273860566491352526706046234450549227581151709314921879592718001940968866" +
 207                 "98683703730220047531433818109270803001720593553052070070607223399946399057131158" +
 208                 "70996357773590271962850611465148375262095653467132900259943976631145459026858989" +
 209                 "79115837093419370441155121920117164880566945938131183843765620627846310490346293" +
 210                 "95002945834116482411496975832601180073169943739350696629571241027323913874175492" +
 211                 "30718624545432220395527352952402459038057445028922468862853365422138157221311632" +
 212                 "88112052146489805180092024719391710555390113943316681515828843687606961102505171" +
 213                 "00739276238555338627255353883096067164466237092264680967125406186950214317621166" +
 214                 "81400975952814939072226011126811531083873176173232352636058381731510345957365382" +
 215                 "23534992935822836851007810884634349983518404451704270189381994243410090575376257" +
 216                 "76757111809008816418331920196262341628816652137471732547772778348877436651882875" +
 217                 "21566857195063719365653903894493664217640031215278702223664636357555035655769488" +
 218                 "86549500270853923617105502131147413744106134445544192101336172996285694899193369" +
 219                 "18472947858072915608851039678195942983318648075608367955149663644896559294818785" +
 220                 "17840387733262470519450504198477420141839477312028158868457072905440575106012852" +
 221                 "58056594703046836344592652552137008068752009593453607316226118728173928074623094" +
 222                 "68536782310609792159936001994623799343421068781349734695924646975250624695861690" +
 223                 "91785739765951993929939955675427146549104568607020990126068187049841780791739240" +
 224                 "71945996323060254707901774527513186809982284730860766536866855516467702911336827" +
 225                 "56310722334672611370549079536583453863719623585631261838715677411873852772292259" +
 226                 "47433737856955384562468010139057278710165129666367644518724656537304024436841408" +
 227                 "14488732957847348490003019477888020460324660842875351848364959195082888323206522" +
 228                 "12810419044804724794929134228495197002260131043006241071797150279343326340799596" +
 229                 "05314460532304885289729176598760166678119379323724538572096075822771784833616135" +
 230                 "82612896226118129455927462767137794487586753657544861407611931125958512655759734" +
 231                 "57301533364263076798544338576171533346232527057200530398828949903425956623297578" +
 232                 "24887350292591668258944568946559926584547626945287805165017206747854178879822768" +
 233                 "06536650641910973434528878338621726156269582654478205672987756426325321594294418" +
 234                 "03994321700009054265076309558846589517170914760743713689331946909098190450129030" +
 235                 "70995662266203031826493657336984195557769637876249188528656866076005660256054457" +
 236                 "11337286840205574416030837052312242587223438854123179481388550075689381124935386" +
 237                 "31863528708379984569261998179452336408742959118074745341955142035172618420084550" +
 238                 "91708456823682008977394558426792142734775608796442792027083121501564063413416171" +
 239                 "66448069815483764491573900121217041547872591998943825364950514771379399147205219" +
 240                 "52907939613762110723849429061635760459623125350606853765142311534966568371511660" +
 241                 "42207963944666211632551577290709784731562782775987881364919512574833287937715714" +
 242                 "59091064841642678309949723674420175862269402159407924480541255360431317992696739" +
 243                 "15754241929660731239376354213923061787675395871143610408940996608947141834069836" +
 244                 "29936753626215452472984642137528910798843813060955526227208375186298370667872244" +
 245                 "30195793793786072107254277289071732854874374355781966511716618330881129120245204" +
 246                 "04868220007234403502544820283425418788465360259150644527165770004452109773558589" +
 247                 "76226554849416217149895323834216001140629507184904277892585527430352213968356790" +
 248                 "18076406042138307308774460170842688272261177180842664333651780002171903449234264" +
 249                 "26629226145600433738386833555534345300426481847398921562708609565062934040526494" +
 250                 "32442614456659212912256488935696550091543064261342526684725949143142393988454324" +
 251                 "86327461842846655985332312210466259890141712103446084271616619001257195870793217" +
 252                 "56969854401339762209674945418540711844643394699016269835160784892451405894094639" +
 253                 "52678073545797003070511636825194877011897640028276484141605872061841852971891540" +
 254                 "19688253289309149665345753571427318482016384644832499037886069008072709327673127" +
 255                 "58196656394114896171683298045513972950668760474091542042842999354102582911350224" +
 256                 "16907694316685742425225090269390348148564513030699251995904363840284292674125734" +
 257                 "22447765584177886171737265462085498294498946787350929581652632072258992368768457" +
 258                 "01782303809656788311228930580914057261086588484587310165815116753332767488701482" +
 259                 "91674197015125597825727074064318086014281490241467804723275976842696339357735429" +
 260                 "30186739439716388611764209004068663398856841681003872389214483176070116684503887" +
 261                 "21236436704331409115573328018297798873659091665961240202177855885487617616198937" +
 262                 "07943800566633648843650891448055710397652146960276625835990519870423001794655367" +
 263                 "9"
 264 }