直角三角形を判定するプログラム

Question

AOJ 直角三角形
３つの正の整数を入力し、それぞれの長さを３辺とする三角形が直角三角形である場合には YES を、違う場合には NO と出力するプログラムを作成して下さい、という問題です。

以下のプログラムを組んだのですが、wrong answer と返されてしまいます。一体何が原因なのでしょうか。
ご教授お願いします。

date_set = gets.chomp.to_f
date_count = 0

loop  do
  break if date_count == date_set
  (a,b,c) = gets.chomp.split.map(&:to_f).sort
  if c.to_f ** 2 == a.to_f ** 2 + b.to_f ** 2
    puts "Yes"
  else
    puts  "No"
  end
  date_count += 1
end

$
1
6 10 8
Yes
$

Answer 1

YESとNOだと思います