Python Separator.classifyの例

コード例 #1

ファイル: TestSeparator.py プロジェクト: migonsu/pyProCT

    def test_classify(self):
        decomposed = {
                    '0': {
                          'traj_A': [0, 1, 2]
                          },
                    '1': {
                          'traj_A': [3],
                          'traj_B': [8, 10]
                          },
                    '2': {
                          'traj_A': [4],
                          'traj_B': [14, 15]
                          },
                    '3': {
                          'traj_A': [5, 6]
                          },
                    '4': {
                          'traj_B': [9, 11, 12, 13, 7]
                          }
                    }

        expected = {
                    'mixed': {
                              '1': {
                                    'traj_A': [3],
                                    'traj_B': [8, 10]
                                    },
                              '2': {
                                    'traj_A': [4],
                                    'traj_B': [14, 15]
                                    }
                              },
                    'pure': {
                             '0': {
                                   'traj_A': [0, 1, 2]
                                   },
                             '3': {
                                   'traj_A': [5, 6]
                                   },
                             '4': {
                                   'traj_B': [9, 11, 12, 13, 7]
                                   }
                             }
                    }
        self.assertDictEqual(expected, Separator.classify(decomposed))

コード例 #2

ファイル: TestSeparator.py プロジェクト: ztypaker/pyProCT

    def test_classify(self):
        decomposed = {
            '0': {
                'traj_A': [0, 1, 2]
            },
            '1': {
                'traj_A': [3],
                'traj_B': [8, 10]
            },
            '2': {
                'traj_A': [4],
                'traj_B': [14, 15]
            },
            '3': {
                'traj_A': [5, 6]
            },
            '4': {
                'traj_B': [9, 11, 12, 13, 7]
            }
        }

        expected = {
            'mixed': {
                '1': {
                    'traj_A': [3],
                    'traj_B': [8, 10]
                },
                '2': {
                    'traj_A': [4],
                    'traj_B': [14, 15]
                }
            },
            'pure': {
                '0': {
                    'traj_A': [0, 1, 2]
                },
                '3': {
                    'traj_A': [5, 6]
                },
                '4': {
                    'traj_B': [9, 11, 12, 13, 7]
                }
            }
        }
        self.assertDictEqual(expected, Separator.classify(decomposed))