Merge pull request #27 from m-chaturvedi/add_templated_headers

cpplint/IWYU: Add atomic, bitset, complex and initializer_list

Author: jwnimmer-tri

cpplint/cpplint.py

@@ -5377,6 +5377,9 @@ def ExpectingFunctionArgs(clean_lines, linenum):
 
 _HEADERS_CONTAINING_TEMPLATES = (
     ('<array>', ('array',)),
+    ('<atomic>', ('atomic',)),
+    ('<bitset>', ('bitset',)),
+    ('<complex>', ('complex',)),
     ('<deque>', ('deque',)),
     ('<functional>', ('unary_function', 'binary_function',
                       'plus', 'minus', 'multiplies', 'divides', 'modulus',
@@ -5395,6 +5398,7 @@ def ExpectingFunctionArgs(clean_lines, linenum):
                       'const_mem_fun_ref_t', 'const_mem_fun1_ref_t',
                       'mem_fun_ref',
                      )),
+    ('<initializer_list>', ('initializer_list',)),
     ('<limits>', ('numeric_limits',)),
     ('<list>', ('list',)),
     ('<map>', ('map', 'multimap',)),

cpplint/cpplint_unittest.py

@@ -1090,6 +1090,50 @@ def testIncludeWhatYouUse(self):
         """,
         'Add #include <array> for array<>'
         '  [build/include_what_you_use] [4]')
+    self.TestIncludeWhatYouUse(
+        """#include <atomic>
+           std::atomic<int> A;
+        """,
+        '')
+    self.TestIncludeWhatYouUse(
+        """#include <string>
+           std::atomic<int> A;
+        """,
+        'Add #include <atomic> for atomic<>'
+        '  [build/include_what_you_use] [4]')
+    self.TestIncludeWhatYouUse(
+        """#include <bitset>
+           std::bitset<4> A;
+        """,
+        '')
+    self.TestIncludeWhatYouUse(
+        """#include <string>
+           std::bitset<4> A;
+        """,
+        'Add #include <bitset> for bitset<>'
+        '  [build/include_what_you_use] [4]')
+    self.TestIncludeWhatYouUse(
+        """#include <complex>
+           std::complex<float> A;
+        """,
+        '')
+    self.TestIncludeWhatYouUse(
+        """#include <string>
+           std::complex<double> A;
+        """,
+        'Add #include <complex> for complex<>'
+        '  [build/include_what_you_use] [4]')
+    self.TestIncludeWhatYouUse(
+        """#include <initializer_list>
+           std::initializer_list<int> A;
+        """,
+        '')
+    self.TestIncludeWhatYouUse(
+        """#include <string>
+           std::initializer_list<int> A;
+        """,
+        'Add #include <initializer_list> for initializer_list<>'
+        '  [build/include_what_you_use] [4]')
     self.TestIncludeWhatYouUse(
         """#include <optional>
            std::optional<int> A;