<commit_before>import unittest
from pymatgen.util.plotting import periodic_table_heatmap
from pymatgen.util.testing import PymatgenTest
import matplotlib


class FuncTestCase(PymatgenTest):

    def test_plot_periodic_heatmap(self):
        random_data = {'Te': 0.11083818874391202, 'Au': 0.7575629917425387,
                       'Th': 1.2475885304040335, 'Ni': -2.0354391922547705}
        plt = periodic_table_heatmap(random_data, cmap="plasma")
        plt = periodic_table_heatmap(random_data)
        plt = periodic_table_heatmap(random_data, max_row=7)
        plt = periodic_table_heatmap(random_data, max_row=10)


if __name__ == "__main__":
    unittest.main()<commit_msg>added unittest<commit_after>import unittest
from pymatgen.util.plotting import periodic_table_heatmap, van_arkel_triangle
from pymatgen.util.testing import PymatgenTest
import matplotlib


class FuncTestCase(PymatgenTest):

    def test_plot_periodic_heatmap(self):
        random_data = {'Te': 0.11083818874391202, 'Au': 0.7575629917425387,
                       'Th': 1.2475885304040335, 'Ni': -2.0354391922547705}
        plt = periodic_table_heatmap(random_data, cmap="plasma")
        plt = periodic_table_heatmap(random_data)
        plt = periodic_table_heatmap(random_data, max_row=7)
        plt = periodic_table_heatmap(random_data, max_row=10)

    def test_van_arkel_triangle(self):
        random_list = [("Fe", "C"), ("Ni", "F")]
        plt = van_arkel_triangle(random_list)
        plt = van_arkel_triangle(random_list, annotate=True)


if __name__ == "__main__":
    unittest.main()
{'fraction_non_alphanumeric': 0.0998104864181933, 'fraction_numerical': 0.09096651926721415, 'mean_word_length': 4.12621359223301, 'pattern_counts': {'":': 2, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '957536', 'n_tokens_mistral': 661, 'n_tokens_neox': 545, 'n_words': 93}