def test_get_vendor_columns(self, expected_vendor, expected_columns, vendor_header, temp_ang_file): expected_columns = ["euler1", "euler2", "euler3", "x", "y" ] + expected_columns n_cols_file = len(expected_columns) temp_ang_file.write(vendor_header) temp_ang_file.close() header = _get_header(open(temp_ang_file.name)) vendor, column_names = _get_vendor_columns(header, n_cols_file) assert vendor == expected_vendor assert column_names == expected_columns
def test_get_vendor_columns_unknown(self, temp_ang_file, n_cols_file): temp_ang_file.write("Look at me!\nI'm Mr. .ang file!\n") temp_ang_file.close() header = _get_header(open(temp_ang_file.name)) with pytest.warns(UserWarning, match=f"Number of columns, {n_cols_file}, "): vendor, column_names = _get_vendor_columns(header, n_cols_file) assert vendor == "unknown" expected_columns = [ "euler1", "euler2", "euler3", "x", "y", "unknown1", "unknown2", "phase_id", ] + ["unknown" + str(i + 3) for i in range(n_cols_file - 8)] assert column_names == expected_columns
def test_get_header(self, temp_ang_file): temp_ang_file.write(ANGFILE_ASTAR_HEADER) temp_ang_file.close() assert _get_header(open(temp_ang_file.name)) == [ "# File created from ACOM RES results", "# ni-dislocations.res", "# ".rstrip(), "# ".rstrip(), "# MaterialName Nickel", "# Formula", "# Symmetry 43", "# LatticeConstants 3.520 3.520 3.520 90.000 90.000 90.000", "# NumberFamilies 4", "# hklFamilies 1 1 1 1 0.000000", "# hklFamilies 2 0 0 1 0.000000", "# hklFamilies 2 2 0 1 0.000000", "# hklFamilies 3 1 1 1 0.000000", "#", "# GRID: SqrGrid#", ]