@pytest.mark.parametrize("shuffle", [True, False])
    def test_X_is_dict(self, BatchIterator, X_dict, shuffle):
        bi = BatchIterator(2, shuffle=shuffle)(X_dict)
        batches = list(bi)
        assert len(batches) == 10

        X0, y0 = batches[0]
        assert X0['one'].shape == (2, 10)
        assert X0['two'].shape == (2, 10)
        assert y0 is None

        Xt1 = np.vstack(b[0]['one'] for b in batches)
        Xt2 = np.vstack(b[0]['two'] for b in batches)
        assert Xt1.shape == X_dict['one'].shape
        assert Xt2.shape == X_dict['two'].shape
        np.testing.assert_equal(Xt1[:, 0], 10 / Xt2[:, 0])

        if shuffle is False:
            np.testing.assert_equal(X_dict['one'][:2], X0['one'])
            np.testing.assert_equal(X_dict['two'][:2], X0['two'])
dataset/ETHPy150Open dnouri/nolearn/nolearn/lasagne/tests/test_base.py/TestBatchIterator.test_X_is_dict/swapped operands of `/`