Exactly. All the class scenes have them in robes and in JK's books, that's all that's required. Any other time they can wear what they want and I don't see why it should be different in the films.