mirror of
https://github.com/bryanthaboi/gen1recomp
synced 2026-10-08 09:42:55 -04:00
bryanthaboi
199 lines
11 KiB
Lua
199 lines
11 KiB
Lua
package.path="./?.lua;./?/init.lua;"..package.path
|
|
love=love or require("tests.love_stub")
|
|
local T=require("tests.harness")
|
|
local Migration=require("src.box.Migration")
|
|
local Items=require("src.box.Items")
|
|
local Store=require("src.box.Store")
|
|
local Catalog=require("src.box.Catalog")
|
|
local Serializer=require("src.core.SaveSerializer")
|
|
local CacheFs=require("src.import.CacheFs")
|
|
local GameVersion=require("src.core.GameVersion")
|
|
local oldRead,tables=CacheFs.readAt,{}
|
|
local function put(v,p,value) tables[GameVersion.cachePrefix(v)..p]=Serializer.encode(value) end
|
|
local base={hp=35,attack=55,defense=30,speed=90,special=50,specialAttack=50,specialDefense=40}
|
|
for _,v in ipairs(GameVersion.ORDER) do
|
|
local gen=GameVersion.generation(v)
|
|
if gen==3 then
|
|
local root="data/generated/gba/pokemon/"
|
|
put(v,root.."names.lua",{[25]="PIKACHU",[151]="MEW",[251]="CELEBI"})
|
|
put(v,root.."national.lua",{toNational={[25]=25,[151]=151,[251]=251},toSpecies={[25]=25,[151]=151,[251]=251}})
|
|
put(v,root.."meta.lua",{[25]={genderRatio=127,growthRate=0},[151]={genderRatio=255,growthRate=0},[251]={genderRatio=255,growthRate=0}})
|
|
put(v,root.."stats.lua",{[25]={hp=35,atk=55,def=40,spe=90,spa=50,spd=50},[151]={hp=100,atk=100,def=100,spe=100,spa=100,spd=100},
|
|
[251]={hp=100,atk=100,def=100,spe=100,spa=100,spd=100}})
|
|
put(v,root.."move_names.lua",{[84]="THUNDERSHOCK"})
|
|
put(v,root.."battle_moves.lua",{moves={[84]={pp=30}}})
|
|
put(v,root.."abilities.lua",{[25]={9,31},[151]={28,0},[251]={30,0}})
|
|
put(v,root.."ability_names.lua",{[9]="STATIC",[28]="SYNCHRONIZE",[30]="NATURAL CURE",[31]="LIGHTNINGROD"})
|
|
put(v,"data/generated/gba/items/pack.lua",{items={[3]={name="GREAT BALL",pocket="POKE_BALLS",importance=0},
|
|
[4]={name="POKé BALL",pocket="POKE_BALLS",importance=0},[200]={name="LEFTOVERS",pocket="ITEMS",importance=0},
|
|
[259]={name="MACH BIKE",pocket="KEY_ITEMS",importance=1},[122]={name="ORANGE MAIL",pocket="ITEMS",importance=0}}})
|
|
else
|
|
local pokemon={PIKACHU={id="PIKACHU",name="PIKACHU",dex=25,growthRate="MEDIUM_FAST",genderRatio=127,catchRate=190,baseStats=base},
|
|
MEW={id="MEW",name="MEW",dex=151,growthRate="MEDIUM_SLOW",genderRatio=255,catchRate=45,baseStats=base}}
|
|
if gen==2 then pokemon.CELEBI={id="CELEBI",name="CELEBI",dex=251,growthRate="MEDIUM_SLOW",genderRatio=255,baseStats=base} end
|
|
put(v,"data/generated/pokemon.lua",pokemon)
|
|
put(v,"data/generated/moves.lua",{THUNDERSHOCK={name="THUNDERSHOCK",pp=30}})
|
|
if gen==1 then
|
|
put(v,"data/generated/items.lua",{POKE_BALL={id="POKE_BALL",name="POKé BALL",index=4},
|
|
BICYCLE={id="BICYCLE",name="BICYCLE",index=6,keyItem=true},TM_01={id="TM_01",name="TM01",index=201}})
|
|
else
|
|
put(v,"data/generated/items.lua",{generation=2,pockets={"ITEM","KEY_ITEM","BALL","TM_HM"},
|
|
POKE_BALL={id="POKE_BALL",name="POKé BALL",index=5,pocket="BALL"},
|
|
LEFTOVERS={id="LEFTOVERS",name="LEFTOVERS",index=0x92,pocket="ITEM"},
|
|
BITTER_BERRY={id="BITTER_BERRY",name="BITTER BERRY",index=0x53,pocket="ITEM"},
|
|
BERRY={id="BERRY",name="BERRY",index=0xad,pocket="ITEM"},
|
|
LIGHT_BALL={id="LIGHT_BALL",name="LIGHT BALL",index=0xa3,pocket="ITEM"},
|
|
TM_DYNAMICPUNCH={id="TM_DYNAMICPUNCH",name="TM01",index=0xbf,pocket="TM_HM"},
|
|
FLOWER_MAIL={id="FLOWER_MAIL",name="FLOWER MAIL",index=0x9e,pocket="ITEM"},
|
|
timeCapsule={[0x19]="LEFTOVERS",[0x2d]="BITTER_BERRY",[0xbe]="BERRY",[0xff]="BERRY"}})
|
|
end
|
|
end
|
|
end
|
|
CacheFs.readAt=function(path) return tables[path] end;Catalog.reset()
|
|
|
|
local function entry(v,id,species,exp)
|
|
local gen=GameVersion.generation(v)
|
|
local mon={species=species or "PIKACHU",nickname="PIKA",ot="ALICE",otName="ALICE",otId=42,level=50,
|
|
exp=exp or 125500,experience=exp or 125500,happiness=150,catchRate=190,
|
|
dvs={attack=7,defense=3,speed=9,special=4},statExp={hp=65535,attack=65535,defense=65535,speed=65535,special=65535},
|
|
moves={{id="THUNDERSHOCK",pp=15,ppUps=0,maxPp=30}}}
|
|
return {id=id or 1,version=v,generation=gen,mon=mon,display=Catalog.describe(v,mon)}
|
|
end
|
|
|
|
for _,from in ipairs({"red","yellow","gold","crystal"}) do
|
|
for k=0,24 do
|
|
local out=assert(Migration.convert(entry(from,k+1,nil,125500+k),"emerald"))
|
|
T.eq(out.nature,k,from.." EXP "..(125500+k).." gives nature "..k)
|
|
T.eq(out.personality%25,k,from.." personality carries the EXP nature")
|
|
end
|
|
end
|
|
|
|
do
|
|
local e=entry("gold",7)
|
|
local a,lines=assert(Migration.convert(e,"firered"))
|
|
local b=assert(Migration.convert(e,"firered"))
|
|
T.eq(Serializer.encode(a.ivs),Serializer.encode(b.ivs),"IV roll is deterministic for one record")
|
|
local perfect=0
|
|
for _,iv in pairs(a.ivs) do
|
|
T.check(iv>=0 and iv<=31,"IV within 0-31")
|
|
if iv==31 then perfect=perfect+1 end
|
|
end
|
|
T.check(perfect>=3,"at least three perfect IVs")
|
|
for key,ev in pairs(a.evs) do T.eq(ev,0,"EV "..key.." reset") end
|
|
T.eq(a.abilityNum,1,"species with a second ability takes the second slot")
|
|
T.eq(a.pokeball,4,"Poké Ball")
|
|
T.eq(a.modernFatefulEncounter,false,"ordinary species has no fateful flag")
|
|
T.eq(lines.natureExp,125500,"preview exposes source EXP for the nature helper")
|
|
local other=assert(Migration.convert(entry("gold",8),"firered"))
|
|
T.check(Serializer.encode(other.ivs)~=Serializer.encode(a.ivs),"different records roll different IVs")
|
|
local shiny=entry("gold",9);shiny.mon.dvs={attack=15,defense=10,speed=10,special=10}
|
|
local s=assert(Migration.convert(shiny,"emerald"))
|
|
T.check(require("src.core.game3.pokemon").isShiny(s),"DV shininess survives with an EXP nature")
|
|
end
|
|
|
|
for _,case in ipairs({{"red","MEW",151},{"gold","MEW",151},{"gold","CELEBI",251}}) do
|
|
local out=assert(Migration.convert(entry(case[1],3,case[2]),"emerald"))
|
|
local perfect=0;for _,iv in pairs(out.ivs) do if iv==31 then perfect=perfect+1 end end
|
|
T.check(perfect>=5,case[2].." from "..case[1].." gets five perfect IVs")
|
|
T.eq(out.abilityNum,0,case[2].." keeps its only ability")
|
|
T.eq(out.modernFatefulEncounter,case[3]==151,case[2].." fateful flag matches the Mew obedience check")
|
|
end
|
|
|
|
local steps=Migration.natureSteps(125503)
|
|
T.eq(steps[1].nature,"Adamant","nature helper lists the current nature first")
|
|
T.eq(steps[1].add,0,"current nature needs no EXP")
|
|
T.eq(steps[2].nature,"Naughty","next nature is one EXP away")
|
|
T.eq(steps[2].add,1,"next nature needs one EXP")
|
|
|
|
do
|
|
local red=entry("red",11);red.mon.catchRate=0x19
|
|
local gold=assert(Migration.convert(red,"gold"))
|
|
T.eq(gold.item,"LEFTOVERS","catch rate 25 becomes LEFTOVERS through the Teru-sama table")
|
|
red.mon.catchRate=0xa3
|
|
T.eq(assert(Migration.convert(red,"crystal")).item,"LIGHT_BALL","Yellow Pikachu catch rate becomes LIGHT BALL")
|
|
red.mon.catchRate=0
|
|
T.eq(assert(Migration.convert(red,"gold")).item,nil,"catch rate 0 holds nothing")
|
|
local held=entry("gold",12);held.mon.item="LEFTOVERS"
|
|
local back=assert(Migration.convert(held,"red"))
|
|
T.eq(back.catchRate,0x92,"Gen 2 held item becomes the Gen 1 catch rate byte")
|
|
local again=entry("red",13);again.mon=back;again.display=Catalog.describe("red",back)
|
|
T.eq(assert(Migration.convert(again,"gold")).item,"LEFTOVERS","held item survives a 2 to 1 to 2 round trip")
|
|
held.mon.item="FLOWER_MAIL";held.mon.mail={message="HI"}
|
|
T.check(not Migration.convert(held,"red"),"mail still refuses Gen 1")
|
|
end
|
|
|
|
do
|
|
local state=Store.new()
|
|
local red={version="red",inventory={POKE_BALL=5,BICYCLE=1,TM_01=1},bagOrder={"POKE_BALL","BICYCLE","TM_01"}}
|
|
local rows=Items.bag(red,"red")
|
|
T.eq(#rows,2,"key items are not offered for storage")
|
|
T.check(not Items.deposit(state,red,"red","BICYCLE",1),"key items refuse storage")
|
|
local s1,save1=assert(Items.deposit(state,red,"red","POKE_BALL",3))
|
|
T.eq(save1.inventory.POKE_BALL,2,"deposit removes from the bag")
|
|
T.eq(Items.locker(s1,"gen12")[1].count,3,"deposit adds to the shared Gen 1 and 2 stash")
|
|
T.check(Store.validate(Serializer.decode(Serializer.encode(s1)))~=nil,"locker survives validation")
|
|
T.check(not Items.deposit(state,red,"red","POKE_BALL",9),"cannot deposit more than the bag has")
|
|
local key=Items.locker(s1,"gen12")[1].key
|
|
local crystal={version="crystal",inventory={}}
|
|
local s2,save2=assert(Items.withdraw(s1,crystal,"crystal",key,2))
|
|
T.eq(save2.inventory.POKE_BALL,2,"Gen 1 balls withdraw into a Gen 2 bag")
|
|
T.eq(Items.locker(s2,"gen12")[1].count,1,"withdraw removes from the stash")
|
|
local s3=assert(Items.deposit(s1,red,"red","TM_01",1))
|
|
local tm
|
|
for _,row in ipairs(Items.locker(s3,"gen12")) do if row.name=="TM01" then tm=row.key end end
|
|
T.check(not Items.withdraw(s3,crystal,"crystal",tm,1),"Gen 1 TM01 does not become Gen 2 TM01")
|
|
local gold={version="gold",inventory={LEFTOVERS=2}}
|
|
local s4=assert(Items.deposit(state,gold,"gold","LEFTOVERS",2))
|
|
local left=Items.locker(s4,"gen12")[1].key
|
|
T.check(not Items.withdraw(s4,{version="red",inventory={}},"red",left,1),"Gen 2-only items refuse a Gen 1 bag")
|
|
|
|
s4.nextId=3
|
|
s4.boxes[1].mons[1]=entry("gold",1)
|
|
s4.boxes[1].mons[2]=entry("red",2)
|
|
local s5=assert(Items.give(s4,{box=1,slot=1},left))
|
|
T.eq(s5.boxes[1].mons[1].mon.item,"LEFTOVERS","give sets the held item")
|
|
T.eq(Items.locker(s5,"gen12")[1].count,1,"give uses one from the stash")
|
|
T.check(not Items.give(s4,{box=1,slot=2},left),"Gen 1 Pokémon cannot hold items")
|
|
local s6=assert(Items.takeHeld(s5,{box=1,slot=1}))
|
|
T.eq(s6.boxes[1].mons[1].mon.item,nil,"take clears the held item")
|
|
T.eq(Items.locker(s6,"gen12")[1].count,2,"take returns the item to the stash")
|
|
local mail=Store.copy(s4);mail.boxes[1].mons[1].mon.item="FLOWER_MAIL"
|
|
T.check(not Items.takeHeld(mail,{box=1,slot=1}),"mail stays on its Pokémon")
|
|
end
|
|
|
|
do
|
|
local state=Store.new()
|
|
local em={version="emerald",bag={pockets={ITEMS={},KEY_ITEMS={{id=259,qty=1}},POKE_BALLS={{id=3,qty=4}},TM_CASE={},BERRY_POUCH={}}}}
|
|
local rows=Items.bag(em,"emerald")
|
|
T.eq(#rows,1,"Gen 3 key items are not offered for storage")
|
|
local s1,save1=Items.deposit(state,em,"emerald",3,2)
|
|
T.check(s1~=nil,"Gen 3 deposit works: "..tostring(save1))
|
|
if s1 then
|
|
T.eq(save1.bag.pockets.POKE_BALLS[1].qty,2,"Gen 3 deposit removes from the ball pocket")
|
|
s1.nextId=2
|
|
local out=assert(Migration.convert(entry("gold",1),"emerald"))
|
|
s1.boxes[1].mons[1]={id=1,version="emerald",generation=3,mon=out,display=Catalog.describe("emerald",out)}
|
|
local key=Items.balls(s1)[1].key
|
|
local s2=assert(Items.swapBall(s1,{box=1,slot=1},key))
|
|
T.eq(s2.boxes[1].mons[1].mon.pokeball,3,"ball swap writes the ball id")
|
|
T.eq(Items.balls(s2)[1].count,1,"ball swap uses exactly one ball")
|
|
T.eq(#Items.balls(s2),1,"the old Poké Ball is not refunded")
|
|
T.check(not Items.swapBall(s2,{box=1,slot=1},key),"same ball is refused")
|
|
local egg=Store.copy(s1);egg.boxes[1].mons[1].mon.isEgg=true
|
|
T.check(not Items.swapBall(egg,{box=1,slot=1},key),"eggs cannot change ball")
|
|
T.check(not Items.swapBall(s1,{box=1,slot=1},"NOPE"),"unknown ball refused")
|
|
end
|
|
end
|
|
|
|
do
|
|
local ctx={save={party={{species="PIKACHU",catchRate=190}}}}
|
|
require("src.script.Commands").set_catch_rate(ctx,1,0xA3)
|
|
T.eq(ctx.save.party[1].catchRate,0xA3,"Yellow starter Pikachu carries the LIGHT BALL catch rate byte")
|
|
local body=assert(io.open("data/scripts/oaks_lab_yellow.lua")):read("*a")
|
|
T.check(body:find('{ "give_pokemon", "PIKACHU", 5 }\n rows[#rows + 1] = { "set_catch_rate", 1, 0xA3 }',1,true)~=nil,
|
|
"Oak's Lab stamps the catch rate right after giving Pikachu")
|
|
end
|
|
|
|
CacheFs.readAt=oldRead;Catalog.reset()
|
|
T.finish("Box bank rules and items")
|