<commit_before>import os
import unittest
import glob

from pymatgen.io.vaspio import Poscar
import pymatgen.command_line.platon_caller as platon

class TestPlatonCallerCase(unittest.TestCase):
    def test_get_space_group(self):
        dir_name = os.path.dirname(__file__)
        name = 'data'
        data_dir_name = os.path.join(dir_name, name, "POSCAR_*")
        for file_name in glob.glob(data_dir_name):
            #space_group =  platon.get_space_group(s)
            number = int(file_name.split('_')[-1])
            poscar = Poscar.from_file(file_name)
            s = poscar.struct
            # SG_NB, SG_HM
            sg_dict =  platon.get_space_group(s)
            self.assertEqual(sg_dict['SG_NB'], number)
<commit_msg>Minor change to formating<commit_after>import os
import unittest
import glob

from pymatgen.io.vaspio import Poscar
import pymatgen.command_line.platon_caller as platon

class TestPlatonCallerCase(unittest.TestCase):
    def test_get_space_group(self):
        dir_name = os.path.dirname(__file__)
        name = 'data'
        data_dir_name = os.path.join(dir_name, name, "POSCAR_*")
        for file_name in glob.glob(data_dir_name):
            number = int(file_name.split('_')[-1])
            poscar = Poscar.from_file(file_name)
            s = poscar.struct
            # SG_NB, SG_HM
            sg_dict =  platon.get_space_group(s)
            self.assertEqual(sg_dict['SG_NB'], number)

{'fraction_non_alphanumeric': 0.0967966573816156, 'fraction_numerical': 0.001392757660167131, 'mean_word_length': 2.832, 'pattern_counts': {'":': 0, '<': 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': '6264941', 'n_tokens_mistral': 495, 'n_tokens_neox': 482, 'n_words': 89}