Add test for flat-square badge