From 58b98082a65ff2076c0555deb2b067c1c01df9c0 Mon Sep 17 00:00:00 2001
From: Lars Nielsen <lnc13.lars@gmail.com>
Date: Sun, 5 Nov 2023 08:38:58 +0100
Subject: [PATCH] Adding has_group and has_channel (#314)

---
 nptdms/tdms.py                | 14 ++++++++++++++
 nptdms/test/test_tdms_file.py | 22 ++++++++++++++++++++++
 2 files changed, 36 insertions(+)

diff --git a/nptdms/tdms.py b/nptdms/tdms.py
index 70bc07b..72a8d27 100644
--- a/nptdms/tdms.py
+++ b/nptdms/tdms.py
@@ -214,6 +214,13 @@ def __len__(self):
         """
         return len(self._groups)
 
+    def __contains__(self, group_name):
+        """ Check if TDMS file contains groupp
+
+        :rtype: Boolean
+        """
+        return group_name in self._groups
+
     def __iter__(self):
         """ Returns an iterator over the names of groups in this file
         """
@@ -355,6 +362,13 @@ def __init__(self, path, properties, channels):
     def __repr__(self):
         return "<TdmsGroup with path %s>" % self.path
 
+    def __contains__(self, channel_name):
+        """ Check if group contains channel
+
+        :rtype: Boolean
+        """
+        return channel_name in self._channels
+
     @property
     def path(self):
         """ Path to the TDMS object for this group
diff --git a/nptdms/test/test_tdms_file.py b/nptdms/test/test_tdms_file.py
index 42238f4..956a430 100644
--- a/nptdms/test/test_tdms_file.py
+++ b/nptdms/test/test_tdms_file.py
@@ -899,6 +899,28 @@ def test_object_repr():
     assert repr(channel) == "<TdmsChannel with path /'Group'/'Channel1'>"
 
 
+def test_group_in_tdms_object():
+    """Test in for TDMS object
+    """
+    test_file = GeneratedFile()
+    test_file.add_segment(*basic_segment())
+    tdms_data = test_file.load()
+
+    assert 'Group' in tdms_data
+    assert 'group' not in tdms_data
+
+
+def test_channel_in_group_object():
+    """Test in for group
+    """
+    test_file = GeneratedFile()
+    test_file.add_segment(*basic_segment())
+    tdms_data = test_file.load()
+
+    assert 'Channel1' in tdms_data['Group']
+    assert 'channel1' not in tdms_data['Group']
+
+
 def test_data_read_from_bytes_io():
     """Test reading data"""