test vs train and queries in train