From 6ec785dc6585e4bafeba18d26c221b73f095fee9 Mon Sep 17 00:00:00 2001 From: Cyprian Laskowski Date: Tue, 12 Oct 2021 18:09:39 +0200 Subject: [PATCH] Implemented eq for Msd and Properties --- conversion_utils/jos_msds_and_properties.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/conversion_utils/jos_msds_and_properties.py b/conversion_utils/jos_msds_and_properties.py index e335f98..9a87a15 100644 --- a/conversion_utils/jos_msds_and_properties.py +++ b/conversion_utils/jos_msds_and_properties.py @@ -208,6 +208,13 @@ class Properties: return 'language={language}, category={category}, lexeme features={lexeme_features}, form_features={form_features}'.\ format(language=self.language, category=self.category, lexeme_features=str(self.lexeme_feature_map), form_features=str(self.form_feature_map)) + def __eq__(self, obj): + return isinstance(obj, Properties)\ + and self.category == obj.category\ + and self.lexeme_feature_map == obj.lexeme_feature_map\ + and self.form_feature_map == obj.form_feature_map\ + and self.language == obj.language + class Msd: """JOS msd.""" @@ -219,6 +226,9 @@ class Msd: def __str__(self): return 'code={code}, language={language}'.format(code=self.code, language=self.language) + def __eq__(self, obj): + return isinstance(obj, Msd) and self.code == obj.code and self.language == obj.language + class Converter: """Converter between Msd and Properties objects."""